mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
fix batch size
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user