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):