Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
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).