File size: 339 Bytes
28b9873 2571544 28b9873 70decc5 2571544 28b9873 70decc5 2571544 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers import GPT2Model, GPT2LMHeadModel
from .configuration_fake_custom import FakeCustomConfig
class FakeCustomModel(GPT2Model):
_auto_class = "AutoModel"
config_class = FakeCustomConfig
class FakeCustomModelForCausalLM(GPT2LMHeadModel):
_auto_class = "AutoModelForCausalLM"
config_class = FakeCustomConfig
|