Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

the example model of 'SGPT-125M-weightedmean-nli-bitfit' not compatible with ST #29

Open
dengwx2009 opened this issue May 14, 2023 · 3 comments

Comments

@dengwx2009
Copy link

hi, I tried to run the example be_st with 'Muennighoff/SGPT-125M-weightedmean-nli-bitfit'
but it throw out the exception as follows:
/usr/local/lib/python3.8/site-packages/sentence_transformers/models/Pooling.py:120 in load TypeError: __init__() got an unexpected keyword argument 'pooling_mode_weightedmean_tokens'
Is ’pooling_mode_weightedmean_tokens‘ in the config.json changed to ’pooling_mode_mean_tokens’ ?

@Muennighoff
Copy link
Owner

Can you try what is explained here: #14 (comment)
You're likely using an old ST version

@MattiL
Copy link

MattiL commented May 14, 2023

I got the same error with the computing of the web interface at https://huggingface.co/bigscience-data/sgpt-bloom-1b7-nli a moment ago.

@Muennighoff
Copy link
Owner

I got the same error with the computing of the web interface at https://huggingface.co/bigscience-data/sgpt-bloom-1b7-nli a moment ago.

Yes that's because the web interface is not using the latest ST version

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants