hf-internal-testing
/

tiny-random-custom-architecture

Model card Files Files and versions Community

File size: 158 Bytes

28b9873

from transformers import GPT2Model, GPT2LMHeadModel

class FakeCustomModel(GPT2Model):
    pass


class FakeCustomModelForCausalLM(GPT2LMHeadModel):
    pass