Skip to content

GitLab

Commit 2dd12e38 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make fix copies again

parent b3911f89
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
View file @ 2dd12e38
...@@ -189,7 +189,7 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline): ...@@ -189,7 +189,7 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline):
else: else:
raise ImportError("Please install accelerate via `pip install accelerate`") 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]: for cpu_offloaded_model in [self.unet, self.text_encoder, self.vae, self.safety_checker]:
if cpu_offloaded_model is not None: if cpu_offloaded_model is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment