From 168f3ccbee1d94a99fa8cc0c1afb9ea7c4d7e865 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 1 Aug 2025 21:24:35 +0300
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index d2dd9fe..2040278 100644
--- a/nodes.py
+++ b/nodes.py
@@ -232,7 +232,7 @@ negative output is meant to be used with NAG, it contains only negative prompt e
                     "prompt_embeds": context,
                     "negative_prompt_embeds": context_null,
                     "echoshot": echoshot,
-                },
+                },{"prompt_embeds": context_null}
             
         t5, = LoadWanVideoT5TextEncoder().loadmodel(model_name, precision, "main_device", quantization)
 
@@ -247,11 +247,8 @@ negative output is meant to be used with NAG, it contains only negative prompt e
         )
         del t5
         mm.soft_empty_cache()
-        gc.collect()
-        negative_only_dict = {
-            "prompt_embeds": prompt_embeds_dict["negative_prompt_embeds"],
-        }
-        return (prompt_embeds_dict, negative_only_dict)
+        gc.collect() 
+        return (prompt_embeds_dict, {"prompt_embeds": prompt_embeds_dict["negative_prompt_embeds"]})
 
 #region TextEncode
 class WanVideoTextEncode: