optimum/habana/transformers/models/gpt2/__init__.py

from .modeling_gpt2 import (
    GaudiGPT2Attention,
    GaudiGPT2Block,
    GaudiGPT2DoubleHeadsModel,
    GaudiGPT2LMHeadModel,
    gaudi_gpt2_forward,
)