chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion]: remove stale offload-manager cleanup in denoising stage (#20587)

Browse Source
This commit is contained in:
Ratish P
2026-03-14 20:26:57 +05:30
committed by GitHub
parent 25e38216b6
commit 574aa2d723
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py
View File

@@ -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