Cannot import name 'Wav2Vec2Processor'

I am trying to use the facebook/data2vec-audio-base-960h model.
As per their model card, this is how to load the model:

 from transformers import Wav2Vec2Processor, Data2VecForCTC

 processor = Wav2Vec2Processor.from_pretrained("facebook/data2vec-audio-base-960h")
 model = Data2VecForCTC.from_pretrained("facebook/data2vec-audio-base-960h")

But I am getting this error:

ImportError                               Traceback (most recent call last)
/tmp/ipython-input-11-2185350118.py in <cell line: 0>()
----> 1 from transformers import Wav2Vec2Processor, Data2VecForCTC
      2 
      3 processor = Wav2Vec2Processor.from_pretrained("facebook/data2vec-audio-base-960h")
      4 model = Data2VecForCTC.from_pretrained("facebook/data2vec-audio-base-960h")

ImportError: cannot import name 'Wav2Vec2Processor' from 'transformers' (/usr/local/lib/python3.11/dist-packages/transformers/__init__.py)

I looked up at stack-overflow: It suggested upgrading the Transformers version.
I did that :

  1. My current Transformers version is 4.53.2
  2. That did not fix. I even upgraded sentence-transformers to 5.0.0
  3. I restarted my session in google colab
    None of them worked.. even tried lowering the version of transformers, but It leads to further dependency conflicts.
    Plz help.
1 Like

It seems that the previous sample on the web was incorrect, and now it works on my Colab.

!pip install -U transformers accelerate huggingface_hub[hf_xet]

#from transformers import Wav2Vec2Processor, Data2VecForCTC
from transformers import Wav2Vec2Processor, Data2VecAudioForCTC

processor = Wav2Vec2Processor.from_pretrained("facebook/data2vec-audio-base-960h")
#model = Data2VecForCTC.from_pretrained("facebook/data2vec-audio-base-960h")
model = Data2VecAudioForCTC.from_pretrained("facebook/data2vec-audio-base-960h")
1 Like

This topic was automatically closed 12 hours after the last reply. New replies are no longer allowed.