Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
cbadedf
1
Parent(s):
77eb8e2
progress-bar
Browse files
app.py
CHANGED
@@ -96,6 +96,7 @@ def generate(
|
|
96 |
num_inference_steps_base: int = 25,
|
97 |
num_inference_steps_refiner: int = 25,
|
98 |
apply_refiner: bool = False,
|
|
|
99 |
) -> PIL.Image.Image:
|
100 |
print(f"** Generating image for: \"{prompt}\" **")
|
101 |
generator = torch.Generator().manual_seed(seed)
|
|
|
96 |
num_inference_steps_base: int = 25,
|
97 |
num_inference_steps_refiner: int = 25,
|
98 |
apply_refiner: bool = False,
|
99 |
+
progress=gr.Progress(track_tqdm=True),
|
100 |
) -> PIL.Image.Image:
|
101 |
print(f"** Generating image for: \"{prompt}\" **")
|
102 |
generator = torch.Generator().manual_seed(seed)
|