src/diffusers/hooks/group_offloading.py · 69cdc25746d880279cb79b2018c7de04b8ecf89f · renzhc / diffusers_dcu

Fix group offloading synchronization bug for parameter-only GroupModule's (#12077) · 69cdc257

Aryan authored Aug 06, 2025



* update

* update

* refactor

* fuck yeah

* make style

* Update src/diffusers/hooks/group_offloading.py
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

* Update src/diffusers/hooks/group_offloading.py

---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

69cdc257

group_offloading.py 41.6 KB

Replace group_offloading.py