fix batch size

This commit is contained in:
Vladimir Mandic
2024-02-05 13:52:37 -05:00
parent 3b4d59877f
commit d74f82c025
2 changed files with 1 additions and 2 deletions
-1
View File
@@ -227,7 +227,6 @@ def process_init(p: StableDiffusionProcessing):
def process_images_inner(p: StableDiffusionProcessing) -> Processed:
"""this is the main loop that both txt2img and img2img use; it calls func_init once inside all the scopes and func_sample once per batch"""
if type(p.prompt) == list:
assert len(p.prompt) > 0
else:
+1 -1
View File
@@ -409,7 +409,7 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):
self.color_corrections.append(processing_helpers.setup_color_correction(image))
processed_images.append(image)
self.init_images = processed_images
self.batch_size = len(self.init_images)
# self.batch_size = len(self.init_images)
if self.overlay_images is not None:
self.overlay_images = self.overlay_images * self.batch_size
if self.color_corrections is not None and len(self.color_corrections) == 1: