Fix Syntax Error, close parenthesis. (#2)
- Fix Syntax Error, close parenthesis. (4204457610c9d68596a993f543b5ebd970ce44f1) Co-authored-by: steven tartakovsky <startakovsky@users.noreply.huggingface.co>
This commit is contained in:
parent
4954caae7f
commit
f8fb469f66
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue