Skip to content

Add support for speech recognition with whisper models.#28

Closed
emascarenhas wants to merge 1 commit into
HabanaAI:habana-mainfrom
emascarenhas:habana-main
Closed

Add support for speech recognition with whisper models.#28
emascarenhas wants to merge 1 commit into
HabanaAI:habana-mainfrom
emascarenhas:habana-main

Conversation

@emascarenhas
Copy link
Copy Markdown

What does this PR do?

It adds support for whisper models for speech recognition. Fine tuning and inference are added. README.md is updated.

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

astachowiczhabana added a commit that referenced this pull request Nov 22, 2024
* Fix clip test

* Skip falcon tests

* Fix clip test

* [SW-209062] Disable default sdpa in Albert (#23)

Transformers' default sdpa implementation caused performance
drop in Albert. Adding Albert to the list of models which don't
yet have sdpa implementation in Gaudi and use eager attention.

* [SW-209210] skip first token in EOS check. (#25) (#27)

* Problem: output of _sample function was filled with padding tokens
   for for bart model.

 * Cause: Bart model uses the same token as decoder_start_token_id and
   end of string.
   See: https://huggingface.co/facebook/bart-large-cnn/blob/main/config.json
   Because of that mechanism filling model output with padding
   tokens after EOS (end of string) toke was replacing whole response
   with padding.

 * Solution: Skip check for EOS for first token in padding filling loop.

* Update CODEOWNERS

* Adding labels clone as workaround to avoid crash (#28)

* [SW-0] Fix style

---------

Co-authored-by: Urszula Golowicz <urszula.golowicz@intel.com>
Co-authored-by: Marcin Łapiński <mlapinskix@habana.ai>
Co-authored-by: Bhargav <beede@habana.ai>
xinyu-intel pushed a commit that referenced this pull request Mar 4, 2025
* Fix clip test

* Skip falcon tests

* Fix clip test

* [SW-209062] Disable default sdpa in Albert (#23)

Transformers' default sdpa implementation caused performance
drop in Albert. Adding Albert to the list of models which don't
yet have sdpa implementation in Gaudi and use eager attention.

* [SW-209210] skip first token in EOS check. (#25) (#27)

* Problem: output of _sample function was filled with padding tokens
   for for bart model.

 * Cause: Bart model uses the same token as decoder_start_token_id and
   end of string.
   See: https://huggingface.co/facebook/bart-large-cnn/blob/main/config.json
   Because of that mechanism filling model output with padding
   tokens after EOS (end of string) toke was replacing whole response
   with padding.

 * Solution: Skip check for EOS for first token in padding filling loop.

* Update CODEOWNERS

* Adding labels clone as workaround to avoid crash (#28)

* [SW-0] Fix style

---------

Co-authored-by: Urszula Golowicz <urszula.golowicz@intel.com>
Co-authored-by: Marcin Łapiński <mlapinskix@habana.ai>
Co-authored-by: Bhargav <beede@habana.ai>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant