From 4334ed58b4be092c488e591abd8e31e4eab9a14b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 20 Jun 2025 15:26:13 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index b7ea0c8..1028837 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3979,6 +3979,8 @@ class WanVideoDecode:
                 enable_vae_tiling = False
             images = vae.decode(latents, device=device, end_=(end_image is not None), tiled=enable_vae_tiling, tile_size=(tile_x//8, tile_y//8), tile_stride=(tile_stride_x//8, tile_stride_y//8))[0]
             vae.model.clear_cache()
+        
+        images = images.cpu()
 
         if normalization == "minmax":
             images = (images - images.min()) / (images.max() - images.min())
@@ -4004,7 +4006,7 @@ class WanVideoDecode:
         vae.to(offload_device)
         mm.soft_empty_cache()
 
-        images = torch.clamp(images.cpu(), 0.0, 1.0)
+        images = torch.clamp(images, 0.0, 1.0)
         images = images.permute(1, 2, 3, 0).float()
 
         return (images,)