Spaces:

sergon19
/

sergon19-green_bg_LoRa10

Runtime error

sergon19 commited on Jun 9, 2024

Commit

43d3c1a

verified ·

1 Parent(s): 59552d5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ pipe = DiffusionPipeline.from_pretrained(
     use_safetensors=True
 )
 pipe.load_lora_weights("sergon19/green_bg_LoRa10")
-_ = pipe.to("cuda")
 def generate_image(prompt):
     image = pipe(prompt=prompt, num_inference_steps=25).images[0]

     use_safetensors=True
 )
 pipe.load_lora_weights("sergon19/green_bg_LoRa10")
+_ = pipe.to("cpu")
 def generate_image(prompt):
     image = pipe(prompt=prompt, num_inference_steps=25).images[0]