Spaces:

Aashi
/

Image_to_Image_SDXL

Runtime error

Aashi commited on Jan 23

Commit

276cb27

•

1 Parent(s): 24d1f9b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,8 +57,8 @@ 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()
 # Define the image generation function
 def generate_image(prompt, image_url):

     "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()
 # Define the image generation function
 def generate_image(prompt, image_url):