This commit is contained in:
Jedrzej Kosinski 2025-08-13 18:19:50 -07:00
parent e7ead85cdb
commit 19db0422f0

View File

@ -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)