xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-08 23:37:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move computed interpolation outside loop

Browse Source
This commit is contained in:
catboxanon 2025-05-01 18:35:12 -04:00
parent ea613ed2e3
commit c544038d6f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes_custom_sampler.py
View File

@ -278,6 +278,7 @@ class ExtendIntermediateSigmas:
# linear space for our interpolation function # linear space for our interpolation function
x = torch.linspace(0, 1, steps + 1, device=sigmas.device)[1:-1] x = torch.linspace(0, 1, steps + 1, device=sigmas.device)[1:-1]
computed_spacing = interpolator(x)
extended_sigmas = [] extended_sigmas = []
for i in range(len(sigmas) - 1): for i in range(len(sigmas) - 1):
@ -287,7 +288,7 @@ class ExtendIntermediateSigmas:
extended_sigmas.append(sigma_current) extended_sigmas.append(sigma_current)
if end_at_sigma <= sigma_current <= start_at_sigma: if end_at_sigma <= sigma_current <= start_at_sigma:
interpolated_steps = interpolator(x) * (sigma_next - sigma_current) + sigma_current interpolated_steps = computed_spacing * (sigma_next - sigma_current) + sigma_current
extended_sigmas.extend(interpolated_steps.tolist()) extended_sigmas.extend(interpolated_steps.tolist())
# Add the last sigma value # Add the last sigma value