本地保存SetFit模型部署预测时遇KeyError:0报错求助

可能的原因及对应解决方案 #

1. Joblib版本不兼容 #

该报错多因joblib版本与模型保存时的版本不匹配导致：

• 卸载当前joblib：

  pip uninstall -y joblib
  

• 安装适配版本（推荐1.2.0，多数SetFit模型兼容此版本）：

  pip install joblib==1.2.0
  

2. 模型文件损坏或下载不完整 #

若从Hub下载模型，可能存在文件缺失或损坏：

• 删除本地缓存的模型文件，路径一般为C:\Users\DELL\.cache\huggingface\hub下对应CyberBullying/cyberbullying-classification-setfit-model-v1的文件夹
• 重新运行加载代码，让程序重新完整下载模型

3. 核心依赖版本不匹配 #

SetFit与Hugging Face相关库版本不适配也会引发问题：

• 升级SetFit到最新稳定版：

  pip install --upgrade setfit
  

• 同步升级相关依赖：

  pip install --upgrade transformers datasets
  

4. 本地模型的正确加载方式（若模型已保存至本地） #

如果是自己保存的本地模型，需传入本地路径而非Hub模型ID：

from setfit import SetFitModel
# 替换为你的本地模型文件夹路径
model = SetFitModel.from_pretrained("./your-local-model-path")
preds = model(["i loved the spiderman movie!", "Another Idiot ...is India really Full of them. It's has nothing to do with BJP / CONgress or Hindu-Muslim, it's about Rich/Poor. You all are so uneducated , but literate ."])
print('========================Predictions========================')
print(preds)

内容的提问来源于stack exchange，提问作者Badi elmaestro