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

from transformers.models.gptj.configuration_gptj import GPTJConfig

from .modeling_gptj import (
    GaudiGPTJAttention,
    GaudiGPTJBlock,
    GaudiGPTJForCausalLM,
    GaudiGPTJModel,
)