Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "37c205eb5d5165b70d3100b599a2bcfc483944f5"
Unverified Commit 86a7845c authored by jonrbates's avatar jonrbates Committed by GitHub
Browse files

Fix typo in speech2text2 doc (#15617) 

Forward looks for inputs, not input_ids
parent 9eb7e9ba
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/model_doc/speech_to_text_2.mdx
View file @ 86a7845c
...@@ -69,7 +69,7 @@ predicted token ids. ...@@ -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)
``` ```
......
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