Spaces:
Running
on
Zero
Running
on
Zero
add gradio progress bar
Browse files- 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')
|