From 909cdf0c990fe9e2a1aeb9eee9567c4ebb8d80b4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lucas=20=20Sk=C3=B6ld?= Date: Sun, 19 May 2024 22:31:43 +0200 Subject: [PATCH] Only change steps if custom scheduler --- src/simple_sampler_custom.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/simple_sampler_custom.py b/src/simple_sampler_custom.py index 6544539..a3ffd55 100644 --- a/src/simple_sampler_custom.py +++ b/src/simple_sampler_custom.py @@ -126,7 +126,7 @@ 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: + if denoise > 0 and scheduler_name not in comfy.samplers.KSampler.SCHEDULERS: steps = int(steps / denoise) if scheduler_name == "align_your_steps": @@ -151,7 +151,6 @@ class SimpleSamplerCustom: sigmas = self.get_denoised_sigmas(VPScheduler.get_sigmas(self, steps, beta_d, beta_min, eps_s)[0], denoise)[1] elif scheduler_name in comfy.samplers.KSampler.SCHEDULERS: - steps = steps / 2 sigmas = BasicScheduler.get_sigmas(self, model, scheduler_name, steps, denoise)[0] else: scheduler_settings = self.scheduler_settings.get(scheduler_name)