From 574aa2d723b105cd206082d493d8aa20dc8d2f50 Mon Sep 17 00:00:00 2001 From: Ratish P <114130421+Ratish1@users.noreply.github.com> Date: Sat, 14 Mar 2026 20:26:57 +0530 Subject: [PATCH] [diffusion]: remove stale offload-manager cleanup in denoising stage (#20587) --- .../runtime/pipelines_core/stages/denoising.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py index 27da4c92b..ba169c85d 100644 --- a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py +++ b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py @@ -736,17 +736,6 @@ class DenoisingStage(PipelineStage): latents, batch ) - offload_mgr = getattr(self.transformer, "_layerwise_offload_manager", None) - if offload_mgr is not None and getattr(offload_mgr, "enabled", False): - offload_mgr.release_all() - - if self.transformer_2 is not None: - offload_mgr_2 = getattr( - self.transformer_2, "_layerwise_offload_manager", None - ) - if offload_mgr_2 is not None and getattr(offload_mgr_2, "enabled", False): - offload_mgr_2.release_all() - # Save STA mask search results if needed if ( not is_warmup