Spaces:
Paused
Paused
Commit
β’
38be318
1
Parent(s):
4c36274
Update app.py
Browse files
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,
|
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,
|
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):
|