primerz commited on
Commit
6b795ea
·
verified ·
1 Parent(s): 00e4337

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -235,7 +235,7 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
235
  del weights_sd
236
  del lora_model
237
 
238
- @spaces.GPU(duration=100)
239
  def generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index, st):
240
  print(loaded_state_dict)
241
  et = time.time()
@@ -302,7 +302,7 @@ def generate_image(prompt, negative, face_emb, face_image, face_kps, image_stren
302
  image=face_image,
303
  strength=1-image_strength,
304
  control_image=images,
305
- num_inference_steps=40,
306
  guidance_scale = guidance_scale,
307
  controlnet_conditioning_scale=[face_strength, depth_control_scale],
308
  ).images[0]
 
235
  del weights_sd
236
  del lora_model
237
 
238
+ @spaces.GPU(duration=120)
239
  def generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index, st):
240
  print(loaded_state_dict)
241
  et = time.time()
 
302
  image=face_image,
303
  strength=1-image_strength,
304
  control_image=images,
305
+ num_inference_steps=50,
306
  guidance_scale = guidance_scale,
307
  controlnet_conditioning_scale=[face_strength, depth_control_scale],
308
  ).images[0]