github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix issue with concatinating advanced prompt embeds.

Browse Source
This commit is contained in:
Jaret Burkett
2026-04-14 16:04:34 -06:00
parent 7eb1226a6d
commit e74bc9ac7b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/advanced_prompt_embeds.py
View File

@@ -139,7 +139,7 @@ class AdvancedPromptEmbeds:
for key in pe.keys(): for key in pe.keys():
if key not in embeds: if key not in embeds:
embeds[key] = [] embeds[key] = []
embeds[key].append(pe[key]) embeds[key].extend(pe[key])
return cls(**embeds) return cls(**embeds)
@classmethod @classmethod
@@ -155,4 +155,4 @@ class AdvancedPromptEmbeds:
f"Cannot split key {key!r}: expected list of length {num_parts}, got {len(values)}" f"Cannot split key {key!r}: expected list of length {num_parts}, got {len(values)}"
) )
for i in range(num_parts): for i in range(num_parts):
split_embeds[i]._store[key] = values[i] split_embeds[i]._store[key] = [values[i]]