12 steps no lora
Browse files
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=
|
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()
|