# from .modeling_my_model import GPT, GPTConfig

from transformers import AutoConfig, AutoModelForCausalLM

from .configuration_my_model import GPTConfig
from .modeling_my_model import GPT

AutoConfig.register("custom_gpt", GPTConfig)
AutoModelForCausalLM.register(GPTConfig, GPT)