__init__.py

# __init__.py

from .model import GPTConfig, GPT

try:
    from transformers import AutoConfig, AutoModel
    AutoConfig.register("gpt", GPTConfig)
    AutoModel.register(GPTConfig, GPT)
except ImportError:
    pass