298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-08 00:07:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix timestep ranges when batch_size > 1

Browse Source
This commit is contained in:
asagi4 2023-07-27 21:14:09 +03:00
parent 4ab75d9cb8
commit 1ea4d84691
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/samplers.py
View File

@ -19,11 +19,11 @@ def sampling_function(model_function, x, timestep, uncond, cond, cond_scale, con
strength = 1.0 strength = 1.0
if 'timestep_start' in cond[1]: if 'timestep_start' in cond[1]:
timestep_start = cond[1]['timestep_start'] timestep_start = cond[1]['timestep_start']
if timestep_in > timestep_start: if timestep_in[0] > timestep_start:
return None return None
if 'timestep_end' in cond[1]: if 'timestep_end' in cond[1]:
timestep_end = cond[1]['timestep_end'] timestep_end = cond[1]['timestep_end']
if timestep_in < timestep_end: if timestep_in[0] < timestep_end:
return None return None
if 'area' in cond[1]: if 'area' in cond[1]:
area = cond[1]['area'] area = cond[1]['area']