From 618a7a3feaebf4652b28c0294666d715124884cb Mon Sep 17 00:00:00 2001 From: GodOfNothing Date: Mon, 24 Mar 2025 18:25:26 +0300 Subject: [PATCH] Added sanity check that upscales is non-decreasing sequence --- comfy_extras/nodes_advanced_samplers.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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})