[S2T] fix example in docs (#10667)

055ed78f · Suraj Patil · GitHub · 89693e17 · 055ed78f
Unverified Commit 055ed78f authored Mar 11, 2021 by Suraj Patil Committed by GitHub Mar 11, 2021
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

docs/source/model_doc/speech_to_text.rst docs/source/model_doc/speech_to_text.rst +4 -4

No files found.
--- a/docs/source/model_doc/speech_to_text.rst
+++ b/docs/source/model_doc/speech_to_text.rst
@@ -68,8 +68,8 @@ be installed as follows: ``apt install libsndfile1-dev``
        >>> ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
        >>> ds = ds.map(map_to_array)
-        >>> input_features = processor(ds["speech"][0], sampling_rate=16_000, return_tensors="pt").input_features  # Batch size 1
+        >>> inputs = processor(ds["speech"][0], sampling_rate=16_000, return_tensors="pt")
-        >>> generated_ids = model.generate(input_ids=input_features)
+        >>> generated_ids = model.generate(input_ids=inputs["input_features"], attention_mask=inputs["attention_mask])
        >>> transcription = processor.batch_decode(generated_ids)
@@ -100,8 +100,8 @@ be installed as follows: ``apt install libsndfile1-dev``
        >>> ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
        >>> ds = ds.map(map_to_array)
-        >>> input_features = processor(ds["speech"][0], sampling_rate=16_000, return_tensors="pt").input_features  # Batch size 1
+        >>> inputs = processor(ds["speech"][0], sampling_rate=16_000, return_tensors="pt")
-        >>> generated_ids = model.generate(input_ids=input_features, forced_bos_token_id=processor.tokenizer.lang_code_to_id["fr"])
+        >>> generated_ids = model.generate(input_ids=inputs["input_features"], attention_mask=inputs["attention_mask], forced_bos_token_id=processor.tokenizer.lang_code_to_id["fr"])
        >>> translation = processor.batch_decode(generated_ids)