Update app.py
Browse files
app.py
CHANGED
@@ -341,7 +341,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
|
|
341 |
return final_image
|
342 |
|
343 |
@spaces.GPU(duration=75)
|
344 |
-
def run_lora(prompt,
|
345 |
if not selected_indices:
|
346 |
raise gr.Error("You must select at least one LoRA before proceeding.")
|
347 |
|
@@ -400,7 +400,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
|
|
400 |
|
401 |
# Generate image
|
402 |
if image_input is not None:
|
403 |
-
final_image = generate_image_to_image(prompt_mash,
|
404 |
yield final_image, seed, gr.update(visible=False)
|
405 |
else:
|
406 |
image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
|
|
|
341 |
return final_image
|
342 |
|
343 |
@spaces.GPU(duration=75)
|
344 |
+
def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
|
345 |
if not selected_indices:
|
346 |
raise gr.Error("You must select at least one LoRA before proceeding.")
|
347 |
|
|
|
400 |
|
401 |
# Generate image
|
402 |
if image_input is not None:
|
403 |
+
final_image = generate_image_to_image(prompt_mash, steps, cfg_scale, width, height, seed)
|
404 |
yield final_image, seed, gr.update(visible=False)
|
405 |
else:
|
406 |
image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
|