[diffusion] fix: fix not respecting dit_layerwise_offload server arg (#16252)

2026-01-13 09:29:07 +08:00
parent 2b42309955
commit 47d485f35f
1 changed files with 4 additions and 1 deletions
--- a/python/sglang/multimodal_gen/runtime/server_args.py
+++ b/python/sglang/multimodal_gen/runtime/server_args.py
@@ -943,7 +943,10 @@ class ServerArgs:

        if not envs.SGLANG_CACHE_DIT_ENABLED:
            # TODO: need a better way to tell this
-            if "wan" in self.pipeline_config.__class__.__name__.lower():
+            if (
+                "wan" in self.pipeline_config.__class__.__name__.lower()
+                and self.dit_layerwise_offload is None
+            ):
                logger.info(
                    "Automatically enable dit_layerwise_offload for Wan for best performance"
                )