Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
0cc4495
1
Parent(s):
10151ae
Update app.py
Browse files
app.py
CHANGED
@@ -322,14 +322,14 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
|
|
322 |
weight_name = sdxl_loras[selected_state_index]["weights"]
|
323 |
|
324 |
full_path_lora = state_dicts[repo_name]["saved_name"]
|
325 |
-
loaded_state_dict = copy.deepcopy(state_dicts[repo_name]["state_dict"])
|
326 |
cross_attention_kwargs = None
|
327 |
et = time.time()
|
328 |
elapsed_time = et - st
|
329 |
print('Small content processing took: ', elapsed_time, 'seconds')
|
330 |
|
331 |
st = time.time()
|
332 |
-
image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name,
|
333 |
return image, gr.update(visible=True)
|
334 |
|
335 |
def shuffle_gallery(sdxl_loras):
|
|
|
322 |
weight_name = sdxl_loras[selected_state_index]["weights"]
|
323 |
|
324 |
full_path_lora = state_dicts[repo_name]["saved_name"]
|
325 |
+
#loaded_state_dict = copy.deepcopy(state_dicts[repo_name]["state_dict"])
|
326 |
cross_attention_kwargs = None
|
327 |
et = time.time()
|
328 |
elapsed_time = et - st
|
329 |
print('Small content processing took: ', elapsed_time, 'seconds')
|
330 |
|
331 |
st = time.time()
|
332 |
+
image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, full_path_lora, lora_scale, sdxl_loras, selected_state_index, st)
|
333 |
return image, gr.update(visible=True)
|
334 |
|
335 |
def shuffle_gallery(sdxl_loras):
|