Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix Long Prompts

Browse Source 
`extend` to `repeat`

Extend only works on singleton dimensions, mia culpa.
This commit is contained in:
AI-Casanova
2024-01-14 15:09:31 -06:00
committed by GitHub
parent 62faa25c44
commit ed9d3d93cc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+1 -1
View File
@@ -152,7 +152,7 @@ def pad_to_same_length(pipe, embeds):
empty_embed = pipe.encode_prompt("", device, 1, False)
max_token_count = max([embed.shape[1] for embed in embeds])
repeats = max_token_count - min([embed.shape[1] for embed in embeds])
empty_batched = empty_embed[0].to(embeds[0].device).expand(embeds[0].shape[0], repeats, -1)
empty_batched = empty_embed[0].to(embeds[0].device).repeat(embeds[0].shape[0], repeats // empty_embed[0].shape[1], 1)
for i, embed in enumerate(embeds):
if embed.shape[1] < max_token_count:
embed = torch.cat([embed, empty_batched], dim=1)