[Whisper] Fix slow tests (#30152)

Browse Source

* fix tests

* style

* more fixes

* move model to device

* move logits to cpu

* update expected values

* use ungated dataset

* fix

* fix

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

...

This commit is contained in:

Sanchit Gandhi

2024-04-19 12:21:46 +01:00

committed by GitHub

parent 91472cf5fc

commit 4ed0e51cc3

1 changed files with 141 additions and 127 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

268

tests/models/whisper/test_modeling_whisper.py

File diff suppressed because one or more lines are too long