Hi, I am trying to convert my model to onnx format with the help of this notebook I got error , since config.json does not exist. My model is a custom model with extra layers, similar to this, Now how can I create a config.json file for this?

You need to subclass it to have the save_pretrained methods available. So instead of class Mean_Pooling_Model(nn.Module): use from transformers.modeling_utils import PreTrainedModel class Mean_Pooling_Model(PreTrainedModel): It will add extra functionality on top of nn.Module. huggingfa…

How to create a config.json after saving a model

🤗Transformers

seanbenhur October 4, 2021, 9:55am 5

Thank you I will try this!

Topic		Replies	Views
Convert Pytorch Model to Huggingface Transformer? 🤗Transformers	2	11328	May 2, 2024
How to get config.json from .pt loaded model? Beginners	1	1364	April 11, 2022
Save custom transformer as PreTrainedModel Intermediate	1	984	September 7, 2021
How to save custom model to get config.json file Beginners	2	1613	October 20, 2024
Saving trained number of epochs in config.json // Custom fields in config.json 🤗Transformers	1	291	November 7, 2022

How to create a config.json after saving a model

Related topics