Update pytorch_model.bin
Hello!
This model should be correctly saved from the model.safetensors. I'll try to see if I can update the pipeline code to user a more recent version of Sentence Transformers, but for now this should be a good fix (I hope!)!
- Tom Aarsen
Appreciate this! Thank you! :)
Oooh!
I used a specific version of the Sentence Transformers (sentence-transformers = {git = "https://github.com/UKPLab/sentence-transformers.git", rev = "ed916fb5efb16a88ad23f9d6bfd9b91bb38dcaf4"}) that implements the selection of revision, but also has the new pooling option pooling_mode_weightedmean_tokens.
So the pipeline's current version of Sentence Transformers fails with __init__() got an unexpected keyword argument 'pooling_mode_weightedmean_tokens'. 😅
We can remove the new ones, they're false by default so it shouldn't make any change other than that it won't crash for older versions.
These two in particular can be removed: https://huggingface.co/avsolatorio/GIST-Embedding-v0/blob/main/1_Pooling/config.json#L7-L8
Apologies for all the mess here!
- Tom Aarsen
Good point! I just did the patch, and it works now. 🎉 Thanks! :D