From f8fb469f66946c92dca5a98e2550b31e2f889b3c Mon Sep 17 00:00:00 2001 From: Sanchit Gandhi Date: Fri, 11 Nov 2022 13:56:34 +0000 Subject: [PATCH] Fix Syntax Error, close parenthesis. (#2) - Fix Syntax Error, close parenthesis. (4204457610c9d68596a993f543b5ebd970ce44f1) Co-authored-by: steven tartakovsky --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index bb37119..044700e 100644 --- a/README.md +++ b/README.md @@ -227,7 +227,7 @@ The "<|en|>" token is used to specify that the speech is in english and should b >>> input_features = processor(ds[0]["audio"]["array"], return_tensors="pt").input_features >>> # Generate logits ->>> logits = model(input_features, decoder_input_ids = torch.tensor([[50258]]).logits +>>> logits = model(input_features, decoder_input_ids = torch.tensor([[50258]])).logits >>> # take argmax and decode >>> predicted_ids = torch.argmax(logits, dim=-1) >>> transcription = processor.batch_decode(predicted_ids)