Enable Wan2.2-S2V modeling to vLLM-omni (#2751)

Signed-off-by: Chendi Xue <chendi.xue@intel.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Chendi.XueandHongsheng Liu authored and GitHub committed 2026-05-06 08:03:48 +08:00
1 parent 353ac8a402
commit 4e41b7bc23
12 files changed
+4188

No files matched your search

+5
View File
@@ -761,6 +761,11 @@ class OmniDiffusionConfig:
self.model_class_name = "NextStep11Pipeline"
self.tf_model_config = TransformerConfig()
self.update_multimodal_support()
elif model_type == "s2v":
if self.model_class_name is None:
self.model_class_name = "WanS2VPipeline"
self.tf_model_config = TransformerConfig()
self.update_multimodal_support()
elif architectures and len(architectures) == 1:
self.model_class_name = architectures[0]
else: