cannot import name 'AutoModel' from 'transformers'

from transformers import AutoModel, AutoTokenizer

image.png

阅读 7.7k
1 个回答

参见:https://github.com/huggingfac...

AutoModel 等价于 TFAutoModel,但是是给 PyTorch 用的。如果您没有安装 PyTorch,这是正常现象。

若您使用的是 TensorFlow,改成导入 TFAutoModel 就好了。

from transformers import TFAutoModel, AutoTokenizer
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题
宣传栏