modeling.py

from transformers import GPT2Model

from .configuration import NewModelConfig

class NewModel(GPT2Model):
    config_class = NewModelConfig

    def __init__(self, config):
        super().__init__(config)