foivospar commited on
Commit
eb9c2d1
1 Parent(s): 2583838

add lcm-lora support

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,11 +113,11 @@ def generate_image(image_path, num_steps, guidance_scale, seed, num_images, use_
113
  if use_lcm:
114
  pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
115
  pipeline.enable_lora()
116
- pipeline = pipeline.to(device)
117
  else:
118
  pipeline.disable_lora()
119
  pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
120
- pipeline = pipeline.to(device)
121
 
122
  if image_path is None:
123
  raise gr.Error(f"Cannot find any input face image! Please upload a face image.")
 
113
  if use_lcm:
114
  pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
115
  pipeline.enable_lora()
116
+ pipeline.to(device)
117
  else:
118
  pipeline.disable_lora()
119
  pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
120
+ pipeline.to(device)
121
 
122
  if image_path is None:
123
  raise gr.Error(f"Cannot find any input face image! Please upload a face image.")