Skip to content

GitLab

Unverified Commit 504ae918 authored by Francisco Kurucz's avatar Francisco Kurucz Committed by GitHub
Browse files

Fix Hubert models in TFHubertModel and TFHubertForCTC documentation code (#20516)

parent 6cb7d6ec
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/transformers/models/hubert/modeling_tf_hubert.py
View file @ 504ae918
......@@ -1457,8 +1457,8 @@ class TFHubertModel(TFHubertPreTrainedModel):
>>> from datasets import load_dataset
>>> import soundfile as sf
>>> processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-base-960h")
>>> model = TFHubertModel.from_pretrained("facebook/hubert-base-960h")
>>> processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-large-ls960-ft")
>>> model = TFHubertModel.from_pretrained("facebook/hubert-large-ls960-ft")
>>> def map_to_array(batch):
......@@ -1583,8 +1583,8 @@ class TFHubertForCTC(TFHubertPreTrainedModel):
>>> from datasets import load_dataset
>>> import soundfile as sf
>>> processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-base-960h")
>>> model = TFHubertForCTC.from_pretrained("facebook/hubert-base-960h")
>>> processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-large-ls960-ft")
>>> model = TFHubertForCTC.from_pretrained("facebook/hubert-large-ls960-ft")
>>> def map_to_array(batch):
......
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