File size: 268 Bytes
28b9873 2571544 28b9873 2571544 28b9873 2571544 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import GPT2Model, GPT2LMHeadModel
from .configuration_fake_custom import FakeCustomConfig
class FakeCustomModel(GPT2Model):
config_class = FakeCustomConfig
class FakeCustomModelForCausalLM(GPT2LMHeadModel):
config_class = FakeCustomConfig
|