AttributeError:类型对象'Wav2Vec2ForCTC‘没有特性'from_pretrained’

Question

我正在尝试微调医学词汇的Wav2Vec2模型。当我试图在我的VS code Jupyter笔记本上运行以下代码时，我得到了一个错误，但当我在Google Colab上运行同样的代码时，它工作得很好。

from transformers import Wav2Vec2ForCTC
 
model = Wav2Vec2ForCTC.from_pretrained(
    "facebook/wav2vec2-base", 
    gradient_checkpointing=True, 
    ctc_loss_reduction="mean", 
    pad_token_id=processor.tokenizer.pad_token_id,
)

这是我在VS代码中得到的错误

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-59-926a81051d7b> in <module>
      1 from transformers import Wav2Vec2ForCTC
      2 
----> 3 model = Wav2Vec2ForCTC.from_pretrained(
      4     "facebook/wav2vec2-base",
      5     gradient_checkpointing=True,

AttributeError: type object 'Wav2Vec2ForCTC' has no attribute 'from_pretrained'

Answer 1

当我卸载这两个库并使用以下命令安装它们各自所需的版本时，问题得到了解决：

pip uninstall huggingface-hub
pip uninstall transformers
pip install huggingface-hub==0.0.8
pip install transformers==4.6.1

在此之后，问题得到了解决。