【已解决】关于transformers模型‘name ‘init_empty_weights‘ is not defined‘报错
·
我在运行 SentenceTransformer 的 all-MiniLM-L6-v2 时报错。
问题根源
在新版本的transformers中(例如我的版本是4.51.0),init_empty_weights已被移动到accelerate库中,而不是直接通过transformers提供,这是hugging face生态重构模块路径的结果。
解决方案
1.安装accelerate库
init_empty_weights现在由accelerate库提供,需要先安装这个库:
在cmd中输入:
pip install accelerate
2.修改导入路径
导入路径添加
from accelerate import init_empty_weights # 新路径
3.检查代码兼容性
如果代码有旧版写法,需要修改为新的,示例:
# 旧版写法(已废弃)
with init_empty_weights():
model = MyModel()
# 新版写法(需配合 accelerate)
from accelerate import init_empty_weights
with init_empty_weights():
model = MyModel()
到此我已经成功解决该报错。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)