mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-07-04 06:17:10 +08:00
Fix pytorch warning. (#8593)
This commit is contained in:
parent
5b12b55e32
commit
91d40086db
@ -31,7 +31,7 @@ def dynamic_slice(
|
|||||||
starts: List[int],
|
starts: List[int],
|
||||||
sizes: List[int],
|
sizes: List[int],
|
||||||
) -> Tensor:
|
) -> 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]
|
return x[slicing]
|
||||||
|
|
||||||
class AttnChunk(NamedTuple):
|
class AttnChunk(NamedTuple):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user