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"
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"