zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-15 06:05:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow custom templates with Ideogram4 TE (#14374)

Browse Source
This commit is contained in:
Jukka Seppänen
2026-06-09 16:11:05 +03:00
committed by GitHub
parent f89999289a
commit 8ed7f458d0
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/text_encoders/ideogram4.py
View File

@@ -32,7 +32,9 @@ class Ideogram4Tokenizer(sd1_clip.SD1Tokenizer):
self.llama_template = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
def tokenize_with_weights(self, text, return_word_ids=False, llama_template=None, **kwargs):
if llama_template is None:
if text.startswith('<|im_start|>'):
llama_text = text
elif llama_template is None:
llama_text = self.llama_template.format(text)
else:
llama_text = llama_template.format(text)