__init__.py

from .configuration_gpt import CustomGPTConfig
from .modeling_gpt import CustomGPT

CustomGPTConfig.register_for_auto_class()
CustomGPT.register_for_auto_class("AutoModelForCausalLM")