tests/pipelines/stable_diffusion/test_stable_diffusion.py · ec1aded12e8fbdc98d6703f9111ed824addebc77 · renzhc / diffusers_dcu

Optimize test files by fixing CPU-offloading usage (#8409) · ec1aded1

Tolga Cangöz authored Jun 06, 2024

* Refactor code to remove unnecessary calls to `to(torch_device)`

* Refactor code to remove unnecessary calls to `to("cuda")`

* Update pipeline_stable_diffusion_diffedit.py

ec1aded1

test_stable_diffusion.py 61.9 KB

Replace test_stable_diffusion.py