fffiloni commited on
Commit
5eae4fe
1 Parent(s): a0b9f4c

add gradio progress bar

Browse files
Files changed (1) hide show
  1. gradio_demo/app.py +1 -1
gradio_demo/app.py CHANGED
@@ -118,7 +118,7 @@ def show_final_preview(preview_row):
118
  @torch.no_grad()
119
  def instantir_restore(
120
  lq, prompt="", steps=30, cfg_scale=7.0, guidance_end=1.0,
121
- creative_restoration=False, seed=3407, height=1024, width=1024, preview_start=0.0):
122
  if creative_restoration:
123
  if "lcm" not in pipe.unet.active_adapters():
124
  pipe.unet.set_adapter('lcm')
 
118
  @torch.no_grad()
119
  def instantir_restore(
120
  lq, prompt="", steps=30, cfg_scale=7.0, guidance_end=1.0,
121
+ creative_restoration=False, seed=3407, height=1024, width=1024, preview_start=0.0, progress=gr.Progress(track_tqdm=True)):
122
  if creative_restoration:
123
  if "lcm" not in pipe.unet.active_adapters():
124
  pipe.unet.set_adapter('lcm')