From f5563853b83aa8fd60484af1c0d6a9691c887d25 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 5 Sep 2022 21:22:30 +0300
Subject: [PATCH] add information about batch to the image

---
 modules/processing.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/modules/processing.py b/modules/processing.py
index a5b2afb9..b744aa87 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -132,7 +132,9 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
             "Sampler": samplers[p.sampler_index].name,
             "CFG scale": p.cfg_scale,
             "Seed": all_seeds[position_in_batch + iteration * p.batch_size],
-            "GFPGAN": ("GFPGAN" if p.use_GFPGAN else None)
+            "GFPGAN": ("GFPGAN" if p.use_GFPGAN else None),
+            "Batch size": (None if p.batch_size < 2 else p.batch_size),
+            "Batch pos": (None if p.batch_size < 2 else position_in_batch),
         }
 
         if p.extra_generation_params is not None: