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

fix IndexError, change callback type

Browse Source
This commit is contained in:
AI-Casanova
2024-11-08 23:31:52 -06:00
parent 34d6d5f92b
commit fb4638288b
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+5 -4
View File
@@ -172,11 +172,12 @@ class PromptEmbedder:
for i in range(self.batchsize):
if len(batch[i]) == 0: # if not using prompt-scheduling, this will be len(batch[i])==1
return None
else:
# causes error in callback
try:
res.append(batch[i][step]) # and this requests element for specific step when called from callback - but self.scheduled_prompt==False so len(batch[i])==1 and step is list index out-of-bounds!
if step != 0: # For Callback
res.append(batch[i][step]) # Diffusers internally doubles batch dimension
except IndexError:
res.append(batch[i][0])
if step != 0: # For Callback
res.append(res[-1]) # Diffusers internally doubles batch dimension
return torch.cat(res)