Force steps to match possible cfg schedule length

2025-05-27 18:06:36 +03:00
parent 934dc127a8
commit 81e7022ab5
1 changed files with 5 additions and 0 deletions
@@ -2312,6 +2312,11 @@ class WanVideoSampler:
        
        steps = int(steps/denoise_strength)

+        if isinstance(cfg, list):
+            if steps != len(cfg):
+                log.info(f"Received {len(cfg)} cfg values, but only {steps} steps. Setting step count to match.")
+                steps = len(cfg)
+
        timesteps = None
        if 'unipc' in scheduler:
            sample_scheduler = FlowUniPCMultistepScheduler(shift=shift)