[diffusion] improve: clamp wanvae decode output in place (#27086)

2026-06-03 10:16:10 +08:00
parent f0e18be0ba
commit 3715a07e87
1 changed files with 1 additions and 1 deletions
@@ -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)