Customize pretrained model for model hub
Hi community,I would like to add mean pooling step inside a customSentenceTransformerclass derived from the modelsentence-transformers/stsb-xlm-r-multilingual, in order to avoid to do this supplementary step after getting the tokens embeddings.My aim is to push this custom model onto model hub. If not using this custom step, it is trivial as below:from transformers import AutoTokenizer, AutoModelSimple export## Instanciate the model
model_name = "sentence-transformers/stsb-xlm-r-multilingual"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name)

## Save the model and tokenizer files into cloned repository
model.save_pretrained("path/to/repo/clone/your-model-name")
tokenizer.save_pretrained("path/to/repo/clone/your-model-name")However, after defining my custom classSentenceTransformerCustomI can’t manage to push on model hub the definition of this class. Do I need to place this custom class definition inside a specific .py file ? Or is there anything to do in order to correctly import this custom class from model hub?Thanks!