From 2fcff772c2c172451b3eead352cd8cf73ad41be3 Mon Sep 17 00:00:00 2001 From: isotr0py <2037008807@qq.com> Date: Sat, 15 Feb 2025 11:46:32 +0800 Subject: [PATCH] fix whisper ci Signed-off-by: isotr0py <2037008807@qq.com> --- tests/models/encoder_decoder/audio_language/test_whisper.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/encoder_decoder/audio_language/test_whisper.py b/tests/models/encoder_decoder/audio_language/test_whisper.py index 80d6897da7e0..5705e17463ed 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(