transformers/tests/models/t5
jiqing-feng 1d7f406e19
fix assisted decoding assistant model inputs (#27503)
* fix assisted decoding attention_cat

* fix attention_mask for assisted decoding

* fix attention_mask len

* fix attn len

* Use a more clean way to prepare assistant models inputs

* fix param meaning

* fix param name

* fix assistant model inputs

* update token type ids

* fix assistant kwargs copy

* add encoder-decoder tests of assisted decoding

* check if assistant kwargs contains updated keys

* revert test

* fix whisper tests

* fix assistant kwargs

* revert whisper test

* delete _extend funcs
2023-11-27 14:23:54 +00:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_t5.py CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
test_modeling_t5.py fix assisted decoding assistant model inputs (#27503) 2023-11-27 14:23:54 +00:00
test_modeling_tf_t5.py Conversation pipeline fixes (#26795) 2023-10-16 17:27:45 +01:00
test_tokenization_t5.py [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00