multimodalart HF staff commited on
Commit
38be318
β€’
1 Parent(s): 4c36274

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -204,7 +204,7 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
204
  del weights_sd
205
  del lora_model
206
  @spaces.GPU
207
- def generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, images, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index):
208
  global last_fused, last_lora
209
  print("Last LoRA: ", last_lora)
210
  print("Current LoRA: ", repo_name)
@@ -319,7 +319,7 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
319
  loaded_state_dict = copy.deepcopy(state_dicts[repo_name]["state_dict"])
320
  cross_attention_kwargs = None
321
 
322
- image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, images, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index)
323
  return image, gr.update(visible=True)
324
 
325
  def shuffle_gallery(sdxl_loras):
 
204
  del weights_sd
205
  del lora_model
206
  @spaces.GPU
207
+ 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):
208
  global last_fused, last_lora
209
  print("Last LoRA: ", last_lora)
210
  print("Current LoRA: ", repo_name)
 
319
  loaded_state_dict = copy.deepcopy(state_dicts[repo_name]["state_dict"])
320
  cross_attention_kwargs = None
321
 
322
+ image = 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)
323
  return image, gr.update(visible=True)
324
 
325
  def shuffle_gallery(sdxl_loras):