File size: 364 Bytes
54f90f2 2571544 28b9873 0b35b1b 2571544 28b9873 0b35b1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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" |