From d138fccf80126a4e82c13310bc74f5add1c701e6 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 26 Feb 2025 11:59:05 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index a3bacbc..45fd74e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -618,6 +618,7 @@ class WanVideoTextEncode:
 
         if force_offload:
             encoder.model.to(offload_device)
+            mm.soft_empty_cache()
 
 
         prompt_embeds_dict = {
@@ -667,6 +668,7 @@ class WanVideoImageClipEncode:
         clip_context = clip.visual(pixel_values)
         if force_offload:
             clip.model.to(offload_device)
+            mm.soft_empty_cache()
 
         aspect_ratio = H / W
         lat_h = round(