wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-15 20:39:48 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Add a guard if no <|im_start|> token is found.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-02-23 22:10:15 +01:00
parent 4fe507248c
commit 6445abe8a0
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy/text_encoders/longcat_image.py
View File

@ -112,10 +112,13 @@ class LongCatImageTEModel(sd1_clip.SD1ClipModel):
template_end = i template_end = i
count_im_start += 1 count_im_start += 1
if out.shape[1] > (template_end + 3): if out.shape[1] > (template_end + 3):
if tok_pairs[template_end + 1][0] == 872: if tok_pairs[template_end + 1][0] == 872:
if tok_pairs[template_end + 2][0] == 198: if tok_pairs[template_end + 2][0] == 198:
template_end += 3 template_end += 3
if template_end == -1:
template_end = 0
suffix_start = None suffix_start = None
for i in range(len(tok_pairs) - 1, -1, -1): for i in range(len(tok_pairs) - 1, -1, -1):