transformers

Fix flaky `test_assisted_decoding_matches_greedy_search`

#35951

Merged