Fix kwargs handling in generate_with_fallback (#29225)

Browse Source

* Fix generate_with_fallback **kwargs

* Change pop to get

* Delete keys from kwargs to prevent overriding generation_config

* Revert to passing kwargs by reference, but make a (shallow) copy

* dict -> copy.copy

* Add test_whisper_longform_multi_batch_beam

...

This commit is contained in:

Ondřej Cífka

2024-04-03 17:51:03 +02:00

committed by GitHub

parent 851f253f4d

commit bcd42c4af9

2 changed files with 63 additions and 2 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

55

tests/models/whisper/test_modeling_whisper.py

File diff suppressed because one or more lines are too long