Add FlaxWhisperForAudioClassification model (#23173)

* Add FlaxWhisperForAudioClassification model * Add models to init * Add models to init * Fix copies * Fix automapping * Fix failing test
2023-05-05 22:53:46 +05:30
parent fc6c8b0eaa
commit 312b104ff6
8 changed files with 395 additions and 7 deletions
--- a/docs/source/en/model_doc/whisper.mdx
+++ b/docs/source/en/model_doc/whisper.mdx
@@ -105,3 +105,9 @@ The original code can be found [here](https://github.com/openai/whisper).

 [[autodoc]] FlaxWhisperForConditionalGeneration
    - __call__
+
+## FlaxWhisperForAudioClassification
+
+[[autodoc]] FlaxWhisperForAudioClassification
+    - __call__
+