Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
370f468
1
Parent(s):
56f0936
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ pipe_hyper.to("cuda")
|
|
57 |
del unet
|
58 |
|
59 |
@spaces.GPU
|
60 |
-
def run_comparison(prompt):
|
61 |
image_turbo=pipe_turbo(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
|
62 |
image_lightning=pipe_lightning(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
|
63 |
image_hyper=pipe_hyper(prompt=prompt, num_inference_steps=1, guidance_scale=0, timesteps=[800]).images[0]
|
|
|
57 |
del unet
|
58 |
|
59 |
@spaces.GPU
|
60 |
+
def run_comparison(prompt, progress=gr.Progress(track_tqdm=True)):
|
61 |
image_turbo=pipe_turbo(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
|
62 |
image_lightning=pipe_lightning(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
|
63 |
image_hyper=pipe_hyper(prompt=prompt, num_inference_steps=1, guidance_scale=0, timesteps=[800]).images[0]
|