wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-27 02:10:08 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Adjust context window

Browse Source
This commit is contained in:
kijai 2026-06-02 18:52:46 +03:00
parent 1085cf2f71
commit d9a28a9b3c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_base.py
View File

@ -1531,8 +1531,7 @@ class WAN21(BaseModel):
out = []
for lat in cond_value.cond:
if lat.ndim > dim and lat.shape[dim] > 1 and lat.shape[dim] == x_in.shape[dim]:
idx = tuple([slice(None)] * dim + [window.index_list])
out.append(lat[idx].to(device))
out.append(window.get_tensor(lat, device, dim=dim, retain_index_list=retain_index_list))
else:
out.append(lat.to(device))
return cond_value._copy_with(out)