fix missing import (#15016)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2022-01-03 19:11:47 +01:00 committed by GitHub
parent 38f95d1846
commit 0b4c3a1a53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1606,6 +1606,7 @@ class Wav2Vec2ForMaskedLM(Wav2Vec2PreTrainedModel):
>>> from transformers import Wav2Vec2Processor, Wav2Vec2ForMaskedLM
>>> from datasets import load_dataset
>>> import soundfile as sf
>>> import torch
>>> processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
>>> model = Wav2Vec2ForMaskedLM.from_pretrained("facebook/wav2vec2-base-960h")