From 14da37cdf0a3c52959310f3235e925a486e9fe98 Mon Sep 17 00:00:00 2001 From: Max Tretikov Date: Thu, 13 Jun 2024 23:49:46 -0600 Subject: [PATCH] Fix possibly-used-before-assignment in samplers.py --- comfy/samplers.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/comfy/samplers.py b/comfy/samplers.py index 9d5f34547..6e7fef748 100644 --- a/comfy/samplers.py +++ b/comfy/samplers.py @@ -701,6 +701,8 @@ def sample(model, noise, positive, negative, cfg, device, sampler, sigmas, model def calculate_sigmas(model_sampling, scheduler_name, steps): + sigmas = None + if scheduler_name == "karras": sigmas = k_diffusion_sampling.get_sigmas_karras(n=steps, sigma_min=float(model_sampling.sigma_min), sigma_max=float(model_sampling.sigma_max)) elif scheduler_name == "exponential": @@ -713,8 +715,10 @@ def calculate_sigmas(model_sampling, scheduler_name, steps): sigmas = ddim_scheduler(model_sampling, steps) elif scheduler_name == "sgm_uniform": sigmas = normal_scheduler(model_sampling, steps, sgm=True) - else: + + if sigmas is None: logging.error("error invalid scheduler {}".format(scheduler_name)) + return sigmas def sampler_object(name):