File size: 206 Bytes
687abcb | 1 2 3 4 5 6 7 8 9 10 | from transformers import GPT2Model
from .configuration import NewModelConfig
class NewModel(GPT2Model):
config_class = NewModelConfig
def __init__(self, config):
super().__init__(config)
|