wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-28 03:12:31 +08:00
Code Issues Actions 14 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 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) return io.NodeOutput(generated_text)