nodes/Clybius-ComfyUI-Extra-Samplers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make intensity modulation default in Supreme sampler, and implement reversible_heun as in its paper.

Browse Source
This commit is contained in:
Clybius
2024-04-03 14:37:28 -05:00
parent ebbbbc4d25
commit a74fa12e92
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra_samplers.py
+2 -2
View File
@@ -1109,7 +1109,7 @@ def sampler_supreme(model, x, sigmas, extra_args=None, callback=None, disable=No
d_i_plus_1 = to_d(x_pred, sigma_i_plus_1, denoised_i_plus_1)
# Update the sample using the Reversible Heun formula
z_k = z_k + dt * (d_i + d_i_plus_1) / 2 - dt**2 * (d_i_plus_1 - d_i) / 4
z_k = z_k + dt * (d_i + d_i_plus_1) / 2 - dt**2 * (d_i_plus_1 - d_i) / 2
case "rkf45": # 6 model calls (expensive)
sigma_i, sigma_i_plus_1 = sigmas[i], sigma_down
dt = sigma_i_plus_1 - sigma_i
@@ -1204,7 +1204,7 @@ def sampler_supreme(model, x, sigmas, extra_args=None, callback=None, disable=No
return x
def sample_supreme(model, x, sigmas, extra_args=None, callback=None, disable=None, s_noise=1., noise_sampler_type="gaussian", noise_sampler=None, eta=1.0, step_method="euler", substep_method="euler", centralization=0.05, normalization=0.05, edge_enhancement=0.25, perphist=0.5, substeps=2, noise_modulation="none", modulation_strength=2.0):
def sample_supreme(model, x, sigmas, extra_args=None, callback=None, disable=None, s_noise=1., noise_sampler_type="gaussian", noise_sampler=None, eta=1.0, step_method="euler", substep_method="euler", centralization=0.05, normalization=0.05, edge_enhancement=0.25, perphist=0.5, substeps=2, noise_modulation="intensity", modulation_strength=2.0):
return sampler_supreme(model, x, sigmas, extra_args=extra_args, callback=callback, disable=disable, s_noise=s_noise, noise_sampler=noise_sampler or get_noise_sampler(x, sigmas, noise_sampler_type, noise_sampler, extra_args), eta=eta, step_method=step_method, substep_method=substep_method, centralization=centralization, normalization=normalization, edge_enhancement=edge_enhancement, perphist=perphist, substeps=substeps, noise_modulation=noise_modulation, modulation_strength=modulation_strength)
# Add your personal samplers below here, just for formatting purposes ;3
nodes.py
+1 -1
View File
@@ -158,7 +158,7 @@ class SamplerSUPREME:
"perphist": ("FLOAT", {"default": 0, "min": -5.0, "max": 5.0, "step":0.01}),
"substeps": ("INT", {"default": 2, "min": 1, "max": 100, "step":1}),
"s_noise": ("FLOAT", {"default": 1, "min": 0.0, "max": 100.0, "step":0.01}),
"noise_modulation": (NOISE_MODULATION_TYPES, ),
"noise_modulation": (NOISE_MODULATION_TYPES, {"default": "intensity"}),
"modulation_strength": ("FLOAT", {"default": 2.0, "min": 0.0, "max": 100.0, "step":0.01}),
}
}