在加载模型时

from sentence_transformers import SentenceTransformer

# 加载模型

sense_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')

报错NameError:name 'init_empty_weights' is not defined

是因为新版本的transformers里面没有init_empty_weights(),可以选择降低版本到

4.49.0

或者

使用accelerate导入init_empty_weights

from accelerate import init_empty_weights

注意,使用jupyter notebook更改导入模块版本后,记得重新启动,读者因为没有重新启动,导致以为更改版本的方法无效,其实是有效的

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐