lsb commited on
Commit
f5b5d99
1 Parent(s): 9b35aea

12 steps no lora

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -30,9 +30,9 @@ inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
30
  safety_checker=None,
31
  ).to(preferred_device)
32
 
33
- inpainting_pipeline.scheduler = LCMScheduler.from_config(inpainting_pipeline.scheduler.config)
34
- inpainting_pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
35
- inpainting_pipeline.fuse_lora()
36
 
37
  working_size = (seg_model_img_size, seg_model_img_size)
38
 
@@ -70,7 +70,7 @@ def app(img, prompt):
70
  image=Image.fromarray(img),
71
  mask_image=(mask),
72
  strength=0.95,
73
- num_inference_steps=8,
74
  ).images[0]
75
  #overlay_img.save("overlay_raw.jpg")
76
  end_time = datetime.now().timestamp()
 
30
  safety_checker=None,
31
  ).to(preferred_device)
32
 
33
+ #inpainting_pipeline.scheduler = LCMScheduler.from_config(inpainting_pipeline.scheduler.config)
34
+ #inpainting_pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
35
+ #inpainting_pipeline.fuse_lora()
36
 
37
  working_size = (seg_model_img_size, seg_model_img_size)
38
 
 
70
  image=Image.fromarray(img),
71
  mask_image=(mask),
72
  strength=0.95,
73
+ num_inference_steps=12,
74
  ).images[0]
75
  #overlay_img.save("overlay_raw.jpg")
76
  end_time = datetime.now().timestamp()