changed return type to generated_tokens

This commit is contained in:
Yousef Rafat 2025-11-20 18:18:01 +02:00
parent 01e93c7b7f
commit 470bd9ef5c

View File

@ -18,7 +18,7 @@ class AutoRegressiveGeneration:
} }
} }
RETURN_TYPES = ("TOKENS",) RETURN_TYPES = ("GENERATED_TOKENS",)
FUNCTION = "generate" FUNCTION = "generate"
CATEGORY = "sampling" CATEGORY = "sampling"
@ -35,7 +35,7 @@ class DecodeTokens:
return { return {
"required": { "required": {
"clip": (IO.CLIP, {"tooltip": "The model used for generation."}), "clip": (IO.CLIP, {"tooltip": "The model used for generation."}),
"tokens": ("TOKENS", ),} "tokens": ("GENERATED_TOKENS", ),}
} }
FUNCTION = "decode" FUNCTION = "decode"