Fix example in Wav2Vec2 documentation (#10096)

* Fix example in Wav2Vec2 documentation * fix style

Fix example in Wav2Vec2 documentation (#10096)
* Fix example in Wav2Vec2 documentation * fix style
4ed76377 · abhishek thakur · GitHub · bf1a06a4 · 4ed76377
Unverified Commit 4ed76377 authored Feb 09, 2021 by abhishek thakur Committed by GitHub Feb 09, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/models/wav2vec2/modeling_wav2vec2.py src/transformers/models/wav2vec2/modeling_wav2vec2.py +2 -1

No files found.
--- 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