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

Commit
1 year ago
cpu_offload: remove all hooks before offload (#7448) * 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>
Author
Parents
Loading