[TTS][Model] support MOSS-TTS series (#3420)

Signed-off-by: Zhang <jianmusings@gmail.com>
Signed-off-by: Zhang Jian <jianmusings@gmail.com>
Signed-off-by: jian <jianmusings@gmail.com>
Signed-off-by: Canlin Guo <961750412@qq.com>
Co-authored-by: Canlin Guo <961750412@qq.com>

vllm_omni/benchmarks/patch/patch.py

@@ -46,6 +46,8 @@ from vllm_omni.benchmarks.data_modules.seed_tts_dataset import (
     SeedTTSSampleRequest,
     SeedTTSTextDataset,
 )
+from vllm_omni.benchmarks.data_modules.sound_effect_dataset import SoundEffectDataset
+from vllm_omni.benchmarks.data_modules.ttsd_dataset import TTSDDataset
 
 _AUDIO_CONTINUITY_THRESHOLD_ENV = "VLLM_OMNI_BENCH_AUDIO_CONTINUITY_THRESHOLD_S"
 _DEFAULT_AUDIO_CONTINUITY_THRESHOLD_S = 0.1
@@ -166,7 +168,13 @@ def get_samples(args, tokenizer):
     is_daily_omni = args.dataset_name == "daily-omni" or (
         args.dataset_name == "hf" and _daily_omni_repo_from_args(args) is not None
     )
-    is_seed_tts = args.dataset_name in ("seed-tts", "seed-tts-text", "seed-tts-design")
+    is_seed_tts = args.dataset_name in (
+        "seed-tts",
+        "seed-tts-text",
+        "seed-tts-design",
+        "ttsd",
+        "sound-effect",
+    )
 
     # Check if we need to handle omni-related backends/datasets
     is_omni_backend = args.backend in ["openai-chat-omni", "openai-audio-speech", "daily-omni"]
@@ -283,6 +291,8 @@ def get_samples(args, tokenizer):
             "seed-tts": SeedTTSDataset,
             "seed-tts-text": SeedTTSTextDataset,
             "seed-tts-design": SeedTTSDesignDataset,
+            "ttsd": TTSDDataset,
+            "sound-effect": SoundEffectDataset,
         }
         DatasetCls = _cls_map[args.dataset_name]
         dataset = DatasetCls(