ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-11 08:20:35 +00:00
Code Issues 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():
if key not in embeds:
embeds[key] = []
embeds[key].append(pe[key])
embeds[key].extend(pe[key])
return cls(**embeds)
@classmethod
@@ -155,4 +155,4 @@ class AdvancedPromptEmbeds:
f"Cannot split key {key!r}: expected list of length {num_parts}, got {len(values)}"
)
for i in range(num_parts):
split_embeds[i]._store[key] = values[i]
split_embeds[i]._store[key] = [values[i]]