[Bugfix][Qwen3-TTS] Fix task type (#1317)

Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
This commit is contained in:
Ekagra Ranjan authored and GitHub committed 2026-02-11 15:49:15 +08:00
1 parent a854f0b7ff
commit dd302b63c4
1 file changed
+1 -1
@@ -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