diffusion: refactor task type of models (#13118)

2025-11-12 12:39:34 +08:00
parent dd909a511c
commit 33d1aeb07f
9 changed files with 96 additions and 71 deletions
--- a/python/sglang/multimodal_gen/runtime/server_args.py
+++ b/python/sglang/multimodal_gen/runtime/server_args.py
@@ -800,7 +800,7 @@ class ServerArgs:

    def check_server_sp_args(self):

-        if self.pipeline_config.is_image_gen:
+        if self.pipeline_config.task_type.is_image_task():
            if (
                (self.sp_degree and self.sp_degree > 1)
                or (self.ulysses_degree and self.ulysses_degree > 1)