File size: 182 Bytes
5fa1a76
 
 
1
2
3
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).