Fix pytorch warning. (#8593)

This commit is contained in:
comfyanonymous 2025-06-19 08:04:52 -07:00 committed by GitHub
parent b7d3b0921c
commit 6cac4144ae

View File

@ -31,7 +31,7 @@ def dynamic_slice(
starts: List[int],
sizes: List[int],
) -> Tensor:
slicing = [slice(start, start + size) for start, size in zip(starts, sizes)]
slicing = tuple(slice(start, start + size) for start, size in zip(starts, sizes))
return x[slicing]
class AttnChunk(NamedTuple):