diff --git a/comfy/context_windows.py b/comfy/context_windows.py index d61efb7e9..702a44d5c 100644 --- a/comfy/context_windows.py +++ b/comfy/context_windows.py @@ -231,7 +231,7 @@ class IndexListContextHandler(ContextHandlerABC): if handled: continue 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)): new_cond_item[cond_key] = window.get_tensor(cond_value, device) # Handle audio_embed (temporal dim is 1)