From d4b79dd0f6034f05b9ee98b847f257764ef5ada8 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 13 Aug 2025 18:27:57 -0700 Subject: [PATCH] Another small fix --- comfy/context_windows.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/context_windows.py b/comfy/context_windows.py index c42e9c1d0..928b111df 100644 --- a/comfy/context_windows.py +++ b/comfy/context_windows.py @@ -132,7 +132,7 @@ class IndexListContextHandler(ContextHandlerABC): cond_item = actual_cond[key] if isinstance(cond_item, torch.Tensor): # check that tensor is the expected length - x.size(0) - if cond_item.ndim < self.dim and cond_item.size(self.dim) == x_in.size(self.dim): + if self.dim < cond_item.ndim and cond_item.size(self.dim) == x_in.size(self.dim): # if so, it's subsetting time - tell controls the expected indeces so they can handle them actual_cond_item = window.get_tensor(cond_item) resized_actual_cond[key] = actual_cond_item.to(device)