"vscode:/vscode.git/clone" did not exist on "59cd9de39da4c406e12c1785f70ee73806ebc6ba"
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
......@@ -69,7 +69,7 @@ predicted token ids.
>>> ds = ds.map(map_to_array)
>>> 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)
```
......
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