Add FlaxWhisperForAudioClassification model (#22883)

* Add FlaxWhisperForAudioClassification model

* Add models to init

* Add models to init

* Fix copies

* Fix automapping
This commit is contained in:
raghavanone
2023-05-04 22:30:16 +05:30
committed by GitHub
parent 3341bb41cd
commit c8f2c5c56e
7 changed files with 390 additions and 3 deletions

View File

@@ -105,3 +105,9 @@ The original code can be found [here](https://github.com/openai/whisper).
[[autodoc]] FlaxWhisperForConditionalGeneration
- __call__
## FlaxWhisperForAudioClassification
[[autodoc]] FlaxWhisperForAudioClassification
- __call__