diff --git a/comfy/context_windows.py b/comfy/context_windows.py index 3ebd0094a..c42e9c1d0 100644 --- a/comfy/context_windows.py +++ b/comfy/context_windows.py @@ -150,7 +150,7 @@ class IndexListContextHandler(ContextHandlerABC): new_cond_item[cond_key] = window.get_tensor(cond_value, device) # if has cond that is a Tensor, check if needs to be subset elif hasattr(cond_value, "cond") and isinstance(cond_value.cond, torch.Tensor): - if cond_value.cond < self.dim and cond_value.cond.size(0) == x_in.size(self.dim): + if cond_value.cond.ndim < self.dim and cond_value.cond.size(0) == x_in.size(self.dim): new_cond_item[cond_key] = cond_value._copy_with(window.get_tensor(cond_value.cond, device)) elif cond_key == "num_video_frames": # for SVD new_cond_item[cond_key] = cond_value._copy_with(cond_value.cond)