wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 04:52:31 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

LTX2 context windows - Add handling for downscaled IC-Lora guide frames

Browse Source
This commit is contained in:
ozbayb 2026-04-07 12:44:28 -06:00
parent f1acd5bd85
commit c9edd2d7c0
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/context_windows.py
View File

@ -198,6 +198,15 @@ def inject_guide_frames_into_window(video_slice: torch.Tensor, window: ContextWi
window.guide_frames_indices = suffix_idx window.guide_frames_indices = suffix_idx
window.guide_overlap_info = overlap_info window.guide_overlap_info = overlap_info
window.guide_kf_local_positions = kf_local_pos window.guide_kf_local_positions = kf_local_pos
# Derive per-overlap-entry latent_downscale_factor from guide entry latent_shape vs guide frame spatial dims.
# guide_frames has full (post-dilation) spatial dims; entry["latent_shape"] has pre-dilation dims.
guide_downscale_factors = []
if guide_frame_count > 0:
full_H = guide_frames.shape[3]
for entry_idx, _ in overlap_info:
entry_H = guide_entries[entry_idx]["latent_shape"][1]
guide_downscale_factors.append(full_H // entry_H)
window.guide_downscale_factors = guide_downscale_factors
if guide_frame_count > 0: if guide_frame_count > 0:
idx = tuple([slice(None)] * dim + [suffix_idx]) idx = tuple([slice(None)] * dim + [suffix_idx])
sliced_guide = guide_frames[idx] sliced_guide = guide_frames[idx]