From 3715a07e87d09959b0dc5c3a34b06b4e8691309e Mon Sep 17 00:00:00 2001 From: Mick Date: Wed, 3 Jun 2026 10:16:10 +0800 Subject: [PATCH] [diffusion] improve: clamp wanvae decode output in place (#27086) --- python/sglang/multimodal_gen/runtime/models/vaes/wanvae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/multimodal_gen/runtime/models/vaes/wanvae.py b/python/sglang/multimodal_gen/runtime/models/vaes/wanvae.py index 2fe771a68..0343754a9 100644 --- a/python/sglang/multimodal_gen/runtime/models/vaes/wanvae.py +++ b/python/sglang/multimodal_gen/runtime/models/vaes/wanvae.py @@ -997,7 +997,7 @@ class AutoencoderKLWan(ParallelTiledVAE): out = unpatchify(out, patch_size=self.config.patch_size) out = out.float() - out = torch.clamp(out, min=-1.0, max=1.0) + out.clamp_(min=-1.0, max=1.0) self.clear_cache() else: out = ParallelTiledVAE.decode(self, z)