gpt-pretrain/custom_models/gpt2/__init__.py

2 lines
54 B
Python

from .modeling_gpt2 import GPT2LMHeadModel, GPT2Model