qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix SamplerEulerCFGpp node.

This commit is contained in:
comfyanonymous 2024-07-02 12:21:08 -04:00
parent 2f03201690
commit 01991f72ce
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_extras/nodes_advanced_samplers.py
View File

@ -60,7 +60,7 @@ from comfy.k_diffusion.sampling import to_d
import comfy.model_patcher import comfy.model_patcher
@torch.no_grad() @torch.no_grad()
def sample_euler_cfgpp(model, x, sigmas, extra_args=None, callback=None, disable=None): def sample_euler_pp(model, x, sigmas, extra_args=None, callback=None, disable=None):
extra_args = {} if extra_args is None else extra_args extra_args = {} if extra_args is None else extra_args
temp = [0] temp = [0]
@ -75,11 +75,11 @@ def sample_euler_cfgpp(model, x, sigmas, extra_args=None, callback=None, disable
for i in trange(len(sigmas) - 1, disable=disable): for i in trange(len(sigmas) - 1, disable=disable):
sigma_hat = sigmas[i] sigma_hat = sigmas[i]
denoised = model(x, sigma_hat * s_in, **extra_args) denoised = model(x, sigma_hat * s_in, **extra_args)
d = to_d(x, sigma_hat, temp[0]) d = to_d(x - denoised + temp[0], sigmas[i], denoised)
if callback is not None: if callback is not None:
callback({'x': x, 'i': i, 'sigma': sigmas[i], 'sigma_hat': sigma_hat, 'denoised': denoised}) callback({'x': x, 'i': i, 'sigma': sigmas[i], 'sigma_hat': sigma_hat, 'denoised': denoised})
dt = sigmas[i + 1] - sigma_hat dt = sigmas[i + 1] - sigma_hat
x = denoised + sigmas[i + 1] * d x = x + d * dt
return x return x
@ -96,10 +96,10 @@ class SamplerEulerCFGpp:
FUNCTION = "get_sampler" FUNCTION = "get_sampler"
def get_sampler(self, version): def get_sampler(self, version):
if version == "regular": if version == "alternative":
sampler = comfy.samplers.KSAMPLER(sample_euler_cfgpp) sampler = comfy.samplers.KSAMPLER(sample_euler_pp)
else: else:
sampler = comfy.samplers.ksampler("euler_pp") sampler = comfy.samplers.ksampler("euler_cfg_pp")
return (sampler, ) return (sampler, )
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {