hf-internal-testing
/

tiny-random-custom-architecture

tiny-random-custom-architecture / modeling_fake_custom.py

Matt

Initial commit

28b9873 6 months ago

158 Bytes

	from transformers import GPT2Model, GPT2LMHeadModel

	class FakeCustomModel(GPT2Model):
	pass


	class FakeCustomModelForCausalLM(GPT2LMHeadModel):
	pass