Akbartus commited on
Commit
a54c9eb
·
1 Parent(s): 55ef5dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]