turn off dit_layerwise_offload for wan on rocm (#17569)

2026-01-23 15:22:42 +08:00
parent 69470dbc1f
commit 2169025b77
3 changed files with 11 additions and 0 deletions
@@ -955,6 +955,7 @@ class ServerArgs:
            if (
                "wan" in self.pipeline_config.__class__.__name__.lower()
                and self.dit_layerwise_offload is None
+                and current_platform.enable_dit_layerwise_offload_for_wan_by_default()
            ):
                logger.info(
                    "Automatically enable dit_layerwise_offload for Wan for best performance"