cannot import name 'AutoModel' from 'transformers'

• Z时代
• 2024-02-21
• 分类：IT

from transformers import AutoModel, AutoTokenizer

回答：

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

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

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

from transformers import TFAutoModel, AutoTokenizer

以上是 cannot import name 'AutoModel' from 'transformers' 的全部内容， 来源链接： utcz.com/p/938395.html