Spaces:

Aashi
/

Image_to_Image_SDXL

Runtime error

Aashi commited on Jan 23

Commit

37b77a6

•

1 Parent(s): 276cb27

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):