From 7b8a6f32349ce88dc8cd91e7be8bf531022a0d90 Mon Sep 17 00:00:00 2001 From: sn0w12 Date: Fri, 17 May 2024 01:49:38 +0200 Subject: [PATCH] Steps are now consistent when changing denoise --- src/simple_sampler_custom.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/simple_sampler_custom.py b/src/simple_sampler_custom.py index 415b513..ad1c1d9 100644 --- a/src/simple_sampler_custom.py +++ b/src/simple_sampler_custom.py @@ -126,6 +126,9 @@ class SimpleSamplerCustom: return SplitSigmasDenoise.get_sigmas(self, sigmas, denoise) def get_custom_sigmas(self, model, model_type, scheduler_name, steps, denoise, unique_id): + if denoise > 0: + steps = int(steps / denoise) + if scheduler_name == "align_your_steps": sigmas = AlignYourStepsScheduler.get_sigmas(self, model_type, steps, denoise)[0] elif scheduler_name == "polyexponential":