wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-22 09:33:29 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Fix cond_value.size() call in context window cond resizing

Browse Source
This commit is contained in:
drozbay 2026-02-25 22:13:31 -07:00
parent d7c1bea5ae
commit 83f11a8db4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/context_windows.py
View File

@ -231,7 +231,7 @@ class IndexListContextHandler(ContextHandlerABC):
if handled: if handled:
continue continue
if isinstance(cond_value, torch.Tensor): if isinstance(cond_value, torch.Tensor):
if (self.dim < cond_value.ndim and cond_value(self.dim) == x_in.size(self.dim)) or \ if (self.dim < cond_value.ndim and cond_value.size(self.dim) == x_in.size(self.dim)) or \
(cond_value.ndim < self.dim and cond_value.size(0) == x_in.size(self.dim)): (cond_value.ndim < self.dim and cond_value.size(0) == x_in.size(self.dim)):
new_cond_item[cond_key] = window.get_tensor(cond_value, device) new_cond_item[cond_key] = window.get_tensor(cond_value, device)
# Handle audio_embed (temporal dim is 1) # Handle audio_embed (temporal dim is 1)