add lcm-lora support

app.py

@@ -58,11 +58,11 @@ pipeline = StableDiffusionPipeline.from_pretrained(
         safety_checker=None
     )
 pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
+pipeline = pipeline.to(device)
 
 # load and disable LCM
 pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
 pipeline.disable_lora()
-pipeline = pipeline.to(device)
 
 def toggle_lcm_ui(value):
     if value:
@@ -113,9 +113,11 @@ def generate_image(image_path, num_steps, guidance_scale, seed, num_images, use_
     if use_lcm:
         pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
         pipeline.enable_lora()
+        pipeline = pipeline.to(device)
     else:
         pipeline.disable_lora()
         pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
+        pipeline = pipeline.to(device)
 
     if image_path is None:
         raise gr.Error(f"Cannot find any input face image! Please upload a face image.")