wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 01:17:24 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Fix slice_cond to correctly handle causal anchor index for temporal offsets

Browse Source
This commit is contained in:
ozbayb 2026-04-25 22:12:47 -06:00
parent d9bace4dcb
commit 8eee52de37
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/context_windows.py
View File

@ -117,7 +117,14 @@ def slice_cond(cond_value, window: IndexListContextWindow, x_in: torch.Tensor, d
# skip leading latent positions that have no corresponding conditioning (e.g. reference frames) # skip leading latent positions that have no corresponding conditioning (e.g. reference frames)
if temporal_offset > 0: if temporal_offset > 0:
indices = [i - temporal_offset for i in window.index_list[temporal_offset:]] anchor_idx = getattr(window, 'causal_anchor_index', None)
if anchor_idx is not None and anchor_idx >= 0:
# anchor occupies one of the no-cond positions, so skip one fewer from window.index_list
skip_count = temporal_offset - 1
else:
skip_count = temporal_offset
indices = [i - temporal_offset for i in window.index_list[skip_count:]]
indices = [i for i in indices if 0 <= i] indices = [i for i in indices if 0 <= i]
else: else:
indices = list(window.index_list) indices = list(window.index_list)