Merge pull request #9445 from gakada/master

Add [batch_number] and [generation_number] filename patterns

1 files changed, 3 insertions, 1 deletions

diff --git a/modules/processing.py b/modules/processing.py
index 8d7b2462..a48fff99 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -683,6 +683,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
                 p.scripts.postprocess_batch(p, x_samples_ddim, batch_number=n)

 

             for i, x_sample in enumerate(x_samples_ddim):

+                p.batch_index = i

+

                 x_sample = 255. * np.moveaxis(x_sample.cpu().numpy(), 0, 2)

                 x_sample = x_sample.astype(np.uint8)

 

@@ -731,7 +733,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
 

                     if opts.return_mask:

                         output_images.append(image_mask)

-                    

+

                     if opts.return_mask_composite:

                         output_images.append(image_mask_composite)