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