Steps are now consistent when changing denoise

2024-05-17 01:49:38 +02:00
parent ac14d7e914
commit 7b8a6f3234
1 changed files with 3 additions and 0 deletions
@@ -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":