src/diffusers/pipelines/pipeline_utils.py · f238cb0736e9daa7d35ad9c7daa2b048d0076aa1 · renzhc / diffusers_dcu

cpu_offload: remove all hooks before offload (#7448) · f238cb07

YiYi Xu authored Mar 28, 2024



* add remove_all_hooks

* a few more fix and tests

* up

* Update src/diffusers/pipelines/pipeline_utils.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* split tests

* add

---------
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

f238cb07

pipeline_utils.py 86.2 KB

Replace pipeline_utils.py