Use first sigma in tensor in denoiser callback if only one exists

Fixes errors in SD Forge when mode is set to uncond.
This commit is contained in:
LoganBooker
2024-07-03 09:49:54 +10:00
committed by GitHub
parent 6f38ff3e9f
commit fa6a0dc9e8

View File

@@ -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