From a6dc8aeb64653f33d29988b571ddca77f58d480d Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 20 Sep 2025 21:52:37 +0300
Subject: [PATCH] offload before vae encode too in WanAnimate loop

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

diff --git a/nodes.py b/nodes.py
index 4688976..1d7cb9a 100644
--- a/nodes.py
+++ b/nodes.py
@@ -4324,6 +4324,9 @@ class WanVideoSampler:
                             self.cache_state = [None, None]
                           
                             if ref_latent is not None:
+                                if offload:
+                                    offload_transformer(transformer)
+                                    offloaded = True
                                 vae.to(device)
                                 if ref_masks is not None:
                                     msk = ref_masks_in[:, start_latent:end_latent].to(device, dtype)