Fix failing tests for XLA generation in TF (#18298)
Fix failing tests for XLA generation in TF (#18298)

* Fix failing test_xla_generate_slow tests

* Fix failing speech-to-text xla_generate tests
huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers