"...resnet50_tensorflow.git" did not exist on "4a0b3e4b1467797e2f0828ad42a2e079769f7e17"
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. ...@@ -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