[Bugfix][Qwen3-TTS] Fix task type (#1317)
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
This commit is contained in:
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
|
||||
|
||||
|
||||
Reference in new issue
Block a user