tests/pipelines/stable_unclip/test_stable_unclip.py · ec1aded12e8fbdc98d6703f9111ed824addebc77 · OpenDAS / diffusers

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_unclip.py 8.23 KB

Replace test_stable_unclip.py