diffusers
[feat] implement `record_stream` when using CUDA streams during group offloading
#11081
Merged
Merged
[feat] implement `record_stream` when using CUDA streams during group offloading #11081
sayakpaul
merged
17 commits into
main
from
record-streams
implement record_stream for better performance.
ffce2d19
fix
f25ea18c
style.
2a28f6df
a-r-r-o-w
commented
on 2025-03-18
merge #11097
41ea4c83
a-r-r-o-w
commented
on 2025-03-18
a-r-r-o-w
commented
on 2025-03-18
resolve conflicts.
f5b69b09
Update src/diffusers/hooks/group_offloading.py
9281e84a
Merge branch 'main' into record-streams
637f84ec
fix conflicts.
612136f8
fixes
d5afea56
Merge branch 'main' into record-streams
fb59f362
Merge branch 'main' into record-streams
4a6eeba6
docstring.
87a93fed
remaining todos in low_cpu_mem_usage
1d4ca615
tests
535dcd1b
sayakpaul
changed the title
DN6
approved these changes
on 2025-04-08
updates to docs.
2ff9112c
Merge branch 'main' into record-streams
b4deedcc
Merge branch 'main' into record-streams
622aba7a
Login to write a write a comment.
Login via GitHub