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

This commit is contained in:
Talmaj Marinc 2026-02-23 22:10:15 +01:00
parent 4fe507248c
commit 6445abe8a0

View File

@ -117,6 +117,9 @@ class LongCatImageTEModel(sd1_clip.SD1ClipModel):
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):
elem = tok_pairs[i][0] elem = tok_pairs[i][0]