| # 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) | |