Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from diffusers import StableDiffusionPipeline
|
|
| 7 |
|
| 8 |
def process_image(prompt, num_inference_steps, guidance_scale):
|
| 9 |
model_path = "Akbartus/Lora360"
|
| 10 |
-
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
|
| 11 |
pipe.unet.load_attn_procs(model_path)
|
| 12 |
pipe.to("cpu")
|
| 13 |
image = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images[0]
|
|
|
|
| 7 |
|
| 8 |
def process_image(prompt, num_inference_steps, guidance_scale):
|
| 9 |
model_path = "Akbartus/Lora360"
|
| 10 |
+
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_safetensors=True)
|
| 11 |
pipe.unet.load_attn_procs(model_path)
|
| 12 |
pipe.to("cpu")
|
| 13 |
image = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images[0]
|