gpt-pretrain/custom_models/gpt2/__init__.py

2 lines
54 B
Python
Raw Normal View History

2023-05-14 22:53:28 +08:00
from .modeling_gpt2 import GPT2LMHeadModel, GPT2Model