from transformers import GPT2Model, GPT2LMHeadModel
from .configuration_fake_custom import FakeCustomConfig

class FakeCustomModel(GPT2Model):
    config_class = FakeCustomConfig


class FakeCustomModelForCausalLM(GPT2LMHeadModel):
    config_class = FakeCustomConfig