This commit is contained in:
Chris
2025-08-05 08:00:54 +10:00
parent c7d61bf2c9
commit a37e627cb8
+3
View File
@@ -84,6 +84,9 @@ class ImageFilter(PreviewImage):
images = torch.stack(list(images[int(i)] for i in images_to_return))
latents = {"samples": torch.stack(list(latents['samples'][int(i)] for i in images_to_return))} if latents is not None else None
masks = torch.stack(list(masks[int(i)] for i in images_to_return)) if masks is not None else None
try: int(pick_list_start)
except: pick_list_start = '0'
return (images, latents, masks, e1, e2, e3, ",".join(str(int(x)+int(pick_list_start)) for x in images_to_return))