add LCM Scheduler

app.py

@@ -53,6 +53,11 @@ pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
     safety_checker=None,
     feature_extractor=None,
 )
+
+# load and disable LCM
+pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
+pipe.disable_lora()
+
 pipe.cuda()
 pipe.load_ip_adapter_instantid(face_adapter)
 pipe.image_proj_model.to("cuda")
@@ -207,9 +212,14 @@ def generate_image(
     seed,
     progress=gr.Progress(track_tqdm=True),
 ):
+
     if prompt is None:
         prompt = "a person"
 
+    # LCM Sceduler Callback
+    pipe.scheduler = diffusers.LCMScheduler.from_config(pipe.scheduler.config)
+    pipe.enable_lora()
+
     # apply the style template
     prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
 
@@ -371,17 +381,17 @@ with gr.Blocks(css=css) as demo:
                 )
                 num_steps = gr.Slider(
                     label="Number of sample steps",
-                    minimum=20,
+                    minimum=2,
                     maximum=100,
                     step=1,
-                    value=30,
+                    value=8,
                 )
                 guidance_scale = gr.Slider(
                     label="Guidance scale",
                     minimum=0.1,
                     maximum=10.0,
                     step=0.1,
-                    value=5,
+                    value=3.0,
                 )
                 seed = gr.Slider(
                     label="Seed",