multimodalart HF staff commited on
Commit
370f468
1 Parent(s): 56f0936

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]