ModuleNotFoundError: No module named 'transformers_modules.SDUIRLab.fuzi'
#1
by Leymore - opened
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("SDUIRLab/fuzi.mingcha-v1.0", trust_remote_code=True)
model = AutoModel.from_pretrained("SDUIRLab/fuzi.mingcha-v1.0", trust_remote_code=True).half().cuda()
有如下报错：
ModuleNotFoundError: No module named 'transformers_modules.SDUIRLab.fuzi'
您好，感谢您指出这个报错。
README.md 里路径名有误，改为模型权重具体的路径即可。因为我们是基于 ChatGLM-6B 做了司法增量训练，所以调用方式与 ChatGLM-6B 相同。README.md 里已经做了修改，您可以再尝试一下。