298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-13 15:03:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added sanity check that upscales is non-decreasing sequence

Browse Source
This commit is contained in:
GodOfNothing 2025-03-24 18:25:26 +03:00
parent c868cb2055
commit 618a7a3fea
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_advanced_samplers.py
View File

@ -102,9 +102,18 @@ class SamplerLCMScalewise:
FUNCTION = "get_sampler" FUNCTION = "get_sampler"
def _validate_upscales(self, upscales):
if not upscales:
return
for i in range(1, len(upscales)):
if upscales[i] < upscales[i-1]:
raise ValueError("`upscales` is expected to be non-decreasing sequence of numbers")
def get_sampler(self, upscales, upscale_method): def get_sampler(self, upscales, upscale_method):
# Turn comma-separated list into string # Turn comma-separated list into string
upscales = [float(value) for value in upscales.split(',')] upscales = [float(value) for value in upscales.split(',')]
self._validate_upscales(upscales)
if len(upscales) == 0: if len(upscales) == 0:
upscales = None upscales = None
sampler = comfy.samplers.KSAMPLER(sample_lcm_scalewise, extra_options={"upscales": upscales, "upscale_method": upscale_method}) sampler = comfy.samplers.KSAMPLER(sample_lcm_scalewise, extra_options={"upscales": upscales, "upscale_method": upscale_method})