Merge branch 'fix_batch_timesteps' of https://github.com/asagi4/ComfyUI

2 years ago · fbf5c51c1c
parent 68be24eead 1ea4d84691
commit fbf5c51c1c
1 changed files with 2 additions and 2 deletions
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@ -19,11 +19,11 @@ def sampling_function(model_function, x, timestep, uncond, cond, cond_scale, con
            strength = 1.0
            if 'timestep_start' in cond[1]:
                timestep_start = cond[1]['timestep_start']
-                if timestep_in > timestep_start:
+                if timestep_in[0] > timestep_start:
                    return None
            if 'timestep_end' in cond[1]:
                timestep_end = cond[1]['timestep_end']
-                if timestep_in < timestep_end:
+                if timestep_in[0] < timestep_end:
                    return None
            if 'area' in cond[1]:
                area = cond[1]['area']