ImportError: cannot import name ‘AdamW’ from ‘transformers’

问题:在LangChain框架中部署RAGatouille库并使用ColBERTv2.0进行检索,然后就会出现ImportError: cannot import name ‘AdamW’ from ‘transformers’ 的报错。

原因:transformers库更新造成的

解决方法:一直在解决各种库之间版本不兼容的问题,最终修改该路径下的代码解决。
envs/lib/python3.10/site-packages/colbert/training/training.py

from transformers import AdamW, get_linear_schedule_with_warmup
from torch.optim import AdamW
from transformers import get_linear_schedule_with_warmup

https://ask.csdn.net/questions/8345762

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐