From e9b537f23c79be6923975a7cb4bb45ef7dc81389 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 2 Aug 2025 19:35:02 +0300
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 8169659..9def415 100644
--- a/nodes.py
+++ b/nodes.py
@@ -241,7 +241,7 @@ of the original Wan templates or a custom system prompt.
 """
 
 
-    def process(self, model_name, precision, positive_prompt, negative_prompt, quantization='disabled', use_disk_cache=True, device="gpu", prompt_extender_args=None):
+    def process(self, model_name, precision, positive_prompt, negative_prompt, quantization='disabled', use_disk_cache=True, device="gpu", extender_args=None):
         from .nodes_model_loading import LoadWanVideoT5TextEncoder
         pbar = ProgressBar(3)
 
@@ -249,8 +249,8 @@ of the original Wan templates or a custom system prompt.
 
         # Handle prompt extension with in-memory cache
         orig_prompt = positive_prompt
-        if prompt_extender_args is not None:
-            extender_key = (orig_prompt, str(prompt_extender_args))
+        if extender_args is not None:
+            extender_key = (orig_prompt, str(extender_args))
             if extender_key in _extender_cache:
                 positive_prompt = _extender_cache[extender_key]
                 log.info(f"Loaded extended prompt from in-memory cache: {positive_prompt}")
@@ -258,16 +258,16 @@ of the original Wan templates or a custom system prompt.
                 from .qwen.qwen import QwenLoader, WanVideoPromptExtender
                 log.info("Using WanVideoPromptExtender to process prompts")
                 qwen, = QwenLoader().load(
-                    prompt_extender_args["model"], 
+                    extender_args["model"], 
                     load_device="main_device" if device == "gpu" else "cpu", 
                     precision=precision)
                 positive_prompt, = WanVideoPromptExtender().generate(
                     qwen=qwen,
-                    max_new_tokens=prompt_extender_args["max_new_tokens"],
+                    max_new_tokens=extender_args["max_new_tokens"],
                     prompt=orig_prompt,
                     device=device,
                     force_offload=False,
-                    custom_system_prompt=prompt_extender_args["system_prompt"],
+                    custom_system_prompt=extender_args["system_prompt"],
                 )
                 log.info(f"Extended positive prompt: {positive_prompt}")
                 _extender_cache[extender_key] = positive_prompt