Likewise, if your NewModel is a subclass of [PreTrainedModel], make sure its | |
config_class attribute is set to the same class you use when registering the model (here | |
NewModelConfig). |
Likewise, if your NewModel is a subclass of [PreTrainedModel], make sure its | |
config_class attribute is set to the same class you use when registering the model (here | |
NewModelConfig). |