diff --git a/tests/models/encoder_decoder/audio_language/test_whisper.py b/tests/models/encoder_decoder/audio_language/test_whisper.py index 80d6897da7e02..5705e17463ed8 100644 --- a/tests/models/encoder_decoder/audio_language/test_whisper.py +++ b/tests/models/encoder_decoder/audio_language/test_whisper.py @@ -104,6 +104,7 @@ def run_test( model=model, tensor_parallel_size=tensor_parallel_size, distributed_executor_backend=distributed_executor_backend, + task="generate", ) sampling_params = SamplingParams(