298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-03 22:07:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix pytorch warning. (#8593)

Browse Source
This commit is contained in:
comfyanonymous 2025-06-19 08:04:52 -07:00 committed by GitHub
parent 5b12b55e32
commit 91d40086db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/modules/sub_quadratic_attention.py
View File

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