使用torch.load加载模型提示No module named 'transformers_modules'要怎么处理
时间: 2023-08-09 10:12:26 浏览: 339
这个错误提示可能是因为您在加载模型时使用了一个旧版本的transformers库,而您的模型中使用了新版本的transformers库中的某些模块。解决方法有两种:
1. 更新transformers库:您可以通过在终端中运行以下命令来更新transformers库:
```
pip install transformers --upgrade
```
然后再尝试加载模型。
2. 修改模型代码:如果您无法更新transformers库,或者更新后仍然无法解决问题,那么您可以尝试修改您的模型代码。具体来说,您需要将模型中使用到的transformers模块的引用修改为新版本的模块名。例如,将:
```
from transformers_modules import *
```
修改为:
```
from transformers import *
```
然后再尝试加载模型。
阅读全文