tests/models/test_modeling_common.py · 4b27c4a494bb07849f8a9a509b2d268bf314f7a7 · renzhc / diffusers_dcu

"vscode:/vscode.git/clone" did not exist on "11629d52686a0f5899db5159bf3357a8e8f1b9da"

[feat] implement `record_stream` when using CUDA streams during group offloading (#11081) · 4b27c4a4

Sayak Paul authored Apr 08, 2025



* implement record_stream for better performance.

* fix

* style.

* merge #11097

* Update src/diffusers/hooks/group_offloading.py
Co-authored-by: Aryan <aryan@huggingface.co>

* fixes

* docstring.

* remaining todos in low_cpu_mem_usage

* tests

* updates to docs.

---------
Co-authored-by: Aryan <aryan@huggingface.co>

4b27c4a4

test_modeling_common.py 81.5 KB

Replace test_modeling_common.py