tests/pipelines/stable_diffusion_2/test_stable_diffusion.py · ec1aded12e8fbdc98d6703f9111ed824addebc77 · OpenDAS / diffusers

"...composable_kernel_rocm.git" did not exist on "54de3e55e1fbd04a7fa218893eb2167d44a9756d"

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 25.4 KB

Replace test_stable_diffusion.py