Update app.py
Browse files
app.py
CHANGED
@@ -355,6 +355,7 @@ def run_lora_multi(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lor
|
|
355 |
appends.append(trigger_word)
|
356 |
prompt_mash = " ".join(prepends + [prompt] + appends)
|
357 |
print("Prompt Mash: ", prompt_mash)
|
|
|
358 |
|
359 |
# Unload previous LoRA weights
|
360 |
with calculateDuration("Unloading LoRA"):
|
@@ -408,9 +409,9 @@ def run_lora_multi(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lor
|
|
408 |
step_counter = 0
|
409 |
for image in image_generator:
|
410 |
step_counter += 1
|
411 |
-
final_image =
|
412 |
progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
|
413 |
-
yield
|
414 |
# yield final_image, seed, gr.update(value=progress_bar, visible=False)
|
415 |
|
416 |
run_lora_multi.zerogpu = True
|
|
|
355 |
appends.append(trigger_word)
|
356 |
prompt_mash = " ".join(prepends + [prompt] + appends)
|
357 |
print("Prompt Mash: ", prompt_mash)
|
358 |
+
print("seed: ", seed)
|
359 |
|
360 |
# Unload previous LoRA weights
|
361 |
with calculateDuration("Unloading LoRA"):
|
|
|
409 |
step_counter = 0
|
410 |
for image in image_generator:
|
411 |
step_counter += 1
|
412 |
+
final_image = images
|
413 |
progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
|
414 |
+
yield images, seed, gr.update(value=progress_bar, visible=True)
|
415 |
# yield final_image, seed, gr.update(value=progress_bar, visible=False)
|
416 |
|
417 |
run_lora_multi.zerogpu = True
|