diff --git a/vllm_omni/model_executor/models/qwen3_tts/qwen3_tts.py b/vllm_omni/model_executor/models/qwen3_tts/qwen3_tts.py index 8514a725d..664d2a295 100644 --- a/vllm_omni/model_executor/models/qwen3_tts/qwen3_tts.py +++ b/vllm_omni/model_executor/models/qwen3_tts/qwen3_tts.py @@ -81,7 +81,7 @@ class Qwen3TTSModelForGeneration(nn.Module): torch_dtype=torch.bfloat16, **attn_kwargs, ) - self.task_type = model_path.split("-")[-1].strip("/") + self.task_type = model_path.split("-")[-1].split("/")[0] # Mark that this model produces multimodal outputs self.have_multimodal_outputs = True