Fix example in Wav2Vec2 documentation (#10096)

* Fix example in Wav2Vec2 documentation * fix style
2021-02-09 12:07:56 +01:00
parent bf1a06a437
commit 4ed763779e
1 changed files with 2 additions and 1 deletions
--- a/src/transformers/models/wav2vec2/modeling_wav2vec2.py
+++ b/src/transformers/models/wav2vec2/modeling_wav2vec2.py
@@ -768,7 +768,8 @@ class Wav2Vec2ForCTC(Wav2Vec2PreTrainedModel):

        Example::

-            >>> from transformers import Wav2Vec2Tokenizer, Wav2Vec2Model
+            >>> import torch
+            >>> from transformers import Wav2Vec2Tokenizer, Wav2Vec2ForCTC
            >>> from datasets import load_dataset
            >>> import soundfile as sf