tests/pipelines/unclip/test_unclip.py · ec1aded12e8fbdc98d6703f9111ed824addebc77 · chenpangpang / 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_unclip.py 16.4 KB

Replace test_unclip.py