make fix copies again

2dd12e38 · Patrick von Platen · b3911f89 · 2dd12e38
Commit 2dd12e38 authored Nov 17, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py ...pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py +1 -1

No files found.
--- a/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
+++ b/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
@@ -189,7 +189,7 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline):
        else:
            raise ImportError("Please install accelerate via `pip install accelerate`")

-        device = torch.device("cuda")
+        device = torch.device(f"cuda:{gpu_id}")

        for cpu_offloaded_model in [self.unet, self.text_encoder, self.vae, self.safety_checker]:
            if cpu_offloaded_model is not None: