Spaces:

Aashi
/

Image_to_Image_SDXL

Runtime error

Aashi commited on Jan 23, 2024

Commit

0331ba4

verified ·

1 Parent(s): 20a4d74

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ pipeline = AutoPipelineForImage2Image.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
 )
 pipeline.load_lora_weights('pytorch_lora_weights_00.safetensors')
-_ = pipeline.to("cuda")
 pipeline.enable_model_cpu_offload()

     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
 )
 pipeline.load_lora_weights('pytorch_lora_weights_00.safetensors')
+# _ = pipeline.to("cuda")
 pipeline.enable_model_cpu_offload()