from transformers import PretrainedModel
from .configuration_fake_custom import FakeCustomConfig

class FakeCustomPreTrainedModel(PreTrainedModel):
    config_class = FakeCustomConfig


class FakeCustomModel(FakeCustomPreTrainedModel):
    _auto_class = "AutoModel"


class FakeCustomForCausalLM(FakeCustomPreTrainedModel):
    _auto_class = "AutoModelForCausalLM"