From 443b710f368dd8d8c64b8ed1caffa480dca53ab2 Mon Sep 17 00:00:00 2001
From: Muerrilla <48160881+muerrilla@users.noreply.github.com>
Date: Mon, 8 Jul 2024 01:02:25 +0330
Subject: [PATCH] Batch handling (Fixes #7, #8)

Correct handling of cond/uncond sigmas for batch_size > 1
---
 scripts/detail_daemon.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/scripts/detail_daemon.py b/scripts/detail_daemon.py
index f42fb42..d63d990 100644
--- a/scripts/detail_daemon.py
+++ b/scripts/detail_daemon.py
@@ -124,6 +124,7 @@ class Script(scripts.Script):
             self.schedule = self.make_schedule(actual_steps, start, end, bias, amount, exponent, start_offset, end_offset, fade, smooth)
             self.mode = mode
             self.cfg_scale = p.cfg_scale
+            self.batch_size = p.batch_size
             on_cfg_denoiser(self.denoiser_callback)              
             self.callback_added = True 
             p.extra_generation_params['Detail Daemon'] = f'mode:{mode},amount:{amount},st:{start},ed:{end},bias:{bias},exp:{exponent},st_offset:{start_offset},ed_offset:{end_offset},fade:{fade},smooth:{1 if smooth else 0}'
@@ -160,9 +161,11 @@ class Script(scripts.Script):
             if idx == 0:
                 tqdm.write(f'\033[33mWARNING:\033[0m Forge does not support `cond` and `uncond` modes, using `both` instead')
         if mode == "cond":
-            params.sigma[0] *= 1 - multiplier
+            for i in range(self.batch_size):
+                params.sigma[i] *= 1 - multiplier
         elif mode == "uncond":
-            params.sigma[1] *= 1 + multiplier
+            for i in range(self.batch_size):
+                params.sigma[self.batch_size + i] *= 1 + multiplier
         else:
             params.sigma *= 1 - multiplier * self.cfg_scale