diff --git a/comfy_extras/nodes_advanced_samplers.py b/comfy_extras/nodes_advanced_samplers.py index 75ed86f78..b4deec04c 100644 --- a/comfy_extras/nodes_advanced_samplers.py +++ b/comfy_extras/nodes_advanced_samplers.py @@ -102,9 +102,18 @@ class SamplerLCMScalewise: 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): # Turn comma-separated list into string upscales = [float(value) for value in upscales.split(',')] + self._validate_upscales(upscales) if len(upscales) == 0: upscales = None sampler = comfy.samplers.KSAMPLER(sample_lcm_scalewise, extra_options={"upscales": upscales, "upscale_method": upscale_method})