Keltezaa commited on
Commit
cc7183b
·
verified ·
1 Parent(s): 6171fd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 = image
412
  progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
413
- yield image, seed, gr.update(value=progress_bar, visible=True)
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