Unverified Commit 0675600a authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Update `WhisperForAudioClassification` doc example (#24188)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent e5dd7432
......@@ -1778,7 +1778,7 @@ class WhisperForAudioClassification(WhisperPreTrainedModel):
>>> predicted_class_ids = torch.argmax(logits).item()
>>> predicted_label = model.config.id2label[predicted_class_ids]
>>> predicted_label
'af_za'
'Afrikaans'
```"""
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
......
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