Skip to content

GitLab

Unverified Commit 0b4c3a1a authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix missing import (#15016) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 38f95d18
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/wav2vec2/modeling_wav2vec2.py
View file @ 0b4c3a1a
...@@ -1606,6 +1606,7 @@ class Wav2Vec2ForMaskedLM(Wav2Vec2PreTrainedModel): ...@@ -1606,6 +1606,7 @@ class Wav2Vec2ForMaskedLM(Wav2Vec2PreTrainedModel):
>>> from transformers import Wav2Vec2Processor, Wav2Vec2ForMaskedLM >>> from transformers import Wav2Vec2Processor, Wav2Vec2ForMaskedLM
>>> from datasets import load_dataset >>> from datasets import load_dataset
>>> import soundfile as sf >>> import soundfile as sf
>>> import torch
>>> processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h") >>> processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
>>> model = Wav2Vec2ForMaskedLM.from_pretrained("facebook/wav2vec2-base-960h") >>> model = Wav2Vec2ForMaskedLM.from_pretrained("facebook/wav2vec2-base-960h")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment