wangyuxinwhy / uniem

unified embedding model
Apache License 2.0
829 stars 64 forks source link

加载预训练模型的问题,是因为transformer版本问题吗 #57

Closed KekeWa closed 1 year ago

KekeWa commented 1 year ago

🐛 bug 说明

用python3.10环境无法加载from_pretrained

image

Python Version

3.10

wangyuxinwhy commented 1 year ago

看起来是 uniem 版本的问题,from_pretrained 是 uniem 0.3.0 以上的 api,用下面命令更新一下版本

python -m pip install -U uniem