From 470bd9ef5c887d01680969a9c610916e97cb5de6 Mon Sep 17 00:00:00 2001
From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com>
Date: Thu, 20 Nov 2025 18:18:01 +0200
Subject: [PATCH] changed return type to generated_tokens

---
 comfy_extras/nodes_autoregressive.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy_extras/nodes_autoregressive.py b/comfy_extras/nodes_autoregressive.py
index 39c5b9952..6dfcaf766 100644
--- a/comfy_extras/nodes_autoregressive.py
+++ b/comfy_extras/nodes_autoregressive.py
@@ -18,7 +18,7 @@ class AutoRegressiveGeneration:
             }
         }
 
-    RETURN_TYPES = ("TOKENS",)
+    RETURN_TYPES = ("GENERATED_TOKENS",)
     FUNCTION = "generate"
 
     CATEGORY = "sampling"
@@ -35,7 +35,7 @@ class DecodeTokens:
         return {
             "required": {
                 "clip": (IO.CLIP, {"tooltip": "The model used for generation."}),
-                "tokens": ("TOKENS", ),}
+                "tokens": ("GENERATED_TOKENS", ),}
         }
 
     FUNCTION = "decode"