SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix typo in speech2text2 doc (#15617)

Browse Source 
Forward looks for inputs, not input_ids
This commit is contained in:
jonrbates
2022-02-15 04:54:34 -08:00
committed by GitHub
parent 9eb7e9ba1d
commit 86a7845c0c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/model_doc/speech_to_text_2.mdx
View File

@@ -69,7 +69,7 @@ predicted token ids.
>>> ds = ds.map(map_to_array) >>> ds = ds.map(map_to_array)
>>> inputs = processor(ds["speech"][0], sampling_rate=16_000, return_tensors="pt") >>> inputs = processor(ds["speech"][0], sampling_rate=16_000, return_tensors="pt")
>>> generated_ids = model.generate(input_ids=inputs["input_values"], attention_mask=inputs["attention_mask"]) >>> generated_ids = model.generate(inputs=inputs["input_values"], attention_mask=inputs["attention_mask"])
>>> transcription = processor.batch_decode(generated_ids) >>> transcription = processor.batch_decode(generated_ids)
``` ```