From fa6a0dc9e818d8607e5ca74b993bbe489dc7afee Mon Sep 17 00:00:00 2001 From: LoganBooker Date: Wed, 3 Jul 2024 09:49:54 +1000 Subject: [PATCH] Use first sigma in tensor in denoiser callback if only one exists Fixes errors in SD Forge when mode is set to uncond. --- scripts/detail_daemon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/detail_daemon.py b/scripts/detail_daemon.py index 480d68c..5b955c1 100644 --- a/scripts/detail_daemon.py +++ b/scripts/detail_daemon.py @@ -157,7 +157,7 @@ class Script(scripts.Script): if self.mode == "cond": params.sigma[0] *= 1 - self.schedule[idx] * .1 elif self.mode == "uncond": - params.sigma[1] *= 1 - self.schedule[idx] * -.1 + params.sigma[1 if params.sigma.size(0) > 1 else 0] *= 1 - self.schedule[idx] * -.1 else: params.sigma *= 1 - self.schedule[idx] * .1 * self.cfg_scale