From d74f82c025883df280f2b8cedac93421e20d539f Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Mon, 5 Feb 2024 13:52:37 -0500 Subject: [PATCH] fix batch size --- modules/processing.py | 1 - modules/processing_class.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/modules/processing.py b/modules/processing.py index 70f3f422e..a4564ef08 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -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: diff --git a/modules/processing_class.py b/modules/processing_class.py index a4c3cfb63..f6465b3ca 100644 --- a/modules/processing_class.py +++ b/modules/processing_class.py @@ -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: