From 86a7845c0c16f24af7e665ff6a7bc3540024a6e4 Mon Sep 17 00:00:00 2001 From: jonrbates Date: Tue, 15 Feb 2022 04:54:34 -0800 Subject: [PATCH] Fix typo in speech2text2 doc (#15617) Forward looks for inputs, not input_ids --- docs/source/model_doc/speech_to_text_2.mdx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/model_doc/speech_to_text_2.mdx b/docs/source/model_doc/speech_to_text_2.mdx index 62509b67ec..72754b67aa 100644 --- a/docs/source/model_doc/speech_to_text_2.mdx +++ b/docs/source/model_doc/speech_to_text_2.mdx @@ -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) ```