I created a model with a custom head. To save the model I created a custom config. But I get the following error: __init__() missing 2 required positional arguments: 'model_type' and 'hidden_size' when I save the model with model.save_pretrained(path) That’s my config class: class ReaderConfig(Pre…

Custom config error when model.save_pretrained

nguyen-brat September 8, 2023, 3:52pm 3

thanks you very helpful

Topic		Replies	Views
Error loading custom model Beginners	0	19	March 11, 2025
Composite Config cannot be saved? 🤗Transformers	1	276	July 12, 2023
Registering custom model and config to AutoModel and AutoConfig Models	1	924	November 7, 2023
Trainer.train() will cause PretrainedConfig default construct Intermediate	1	222	February 29, 2024
Config parameters for custom models 🤗Transformers	0	109	April 21, 2024