File size: 215 Bytes
d16379a |
1 2 3 4 5 6 7 8 9 10 |
# __init__.py
from .model import GPTConfig, GPT
try:
from transformers import AutoConfig, AutoModel
AutoConfig.register("gpt", GPTConfig)
AutoModel.register(GPTConfig, GPT)
except ImportError:
pass |