From 8d9dc98fbab0b7bee7100063db6fd9515ce9285c Mon Sep 17 00:00:00 2001 From: chaObserv <154517000+chaObserv@users.noreply.github.com> Date: Mon, 6 Jan 2025 19:35:31 +0800 Subject: [PATCH] Remove more space --- comfy/k_diffusion/sampling.py | 2 +- comfy_extras/nodes_custom_sampler.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/k_diffusion/sampling.py b/comfy/k_diffusion/sampling.py index 4c8d18f2..446c9f7b 100644 --- a/comfy/k_diffusion/sampling.py +++ b/comfy/k_diffusion/sampling.py @@ -1217,7 +1217,7 @@ def sample_sa_solver(model, x, sigmas, extra_args=None, callback=None, disable=F noise=0, sigma=sigmas[-1]) @torch.no_grad() -def sample_sa_solver_pece(model, x, sigmas, extra_args=None, callback=None, disable=False, predictor_order=3, corrector_order=4, tau_func=None, noise_sampler=None): +def sample_sa_solver_pece(model, x, sigmas, extra_args=None, callback=None, disable=False, predictor_order=3, corrector_order=4, tau_func=None, noise_sampler=None): if len(sigmas) <= 1: return x return sample_sa_solver(model, x, sigmas, extra_args=extra_args, callback=callback, disable=disable, diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py index 3cffdf8a..d8094e12 100644 --- a/comfy_extras/nodes_custom_sampler.py +++ b/comfy_extras/nodes_custom_sampler.py @@ -413,7 +413,7 @@ class SamplerDPMAdaptative: "icoeff": icoeff, "dcoeff": dcoeff, "accept_safety": accept_safety, "eta": eta, "s_noise":s_noise }) return (sampler, ) - + class SamplerSASolver: @classmethod def INPUT_TYPES(s):