wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-27 10:52:31 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Translate think tokens

Browse Source
This commit is contained in:
kijai 2026-04-12 21:25:19 +03:00
parent 387e8d8a4c
commit c857b6c657
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy_extras/nodes_textgen.py
View File

@ -71,7 +71,15 @@ class TextGenerate(io.ComfyNode):
seed=seed
)
generated_text = clip.decode(generated_ids, skip_special_tokens=True)
generated_text = clip.decode(generated_ids, skip_special_tokens=not thinking)
if thinking:
# Translate Gemma4 thinking channel markers to standard <think>/</think> tags
generated_text = generated_text.replace("<|channel>thought\n", "<think>\n")
generated_text = generated_text.replace("<channel|>", "</think>")
# Strip remaining special tokens
generated_text = generated_text.replace("<turn|>", "").replace("<eos>", "").strip()
return io.NodeOutput(generated_text)