fffiloni commited on
Commit
ff6eb27
·
verified ·
1 Parent(s): 5f523d5

lower zerogpu duration requirement + display tqdm tracking

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ pipeline = StableDiffusionXLPipeline.from_pretrained(
23
  pipeline.enable_model_cpu_offload()
24
  pipeline.enable_vae_slicing()
25
 
26
- @spaces.GPU(duration=120)
27
- def run(ref_path, ref_style, ref_prompt, prompt1):
28
  # DDIM inversion
29
  src_style = f"{ref_style}"
30
  src_prompt = f"{ref_prompt}, {src_style}."
 
23
  pipeline.enable_model_cpu_offload()
24
  pipeline.enable_vae_slicing()
25
 
26
+ @spaces.GPU(duration=60)
27
+ def run(ref_path, ref_style, ref_prompt, prompt1, progress=gr.Progress(track_tqdm=True)):
28
  # DDIM inversion
29
  src_style = f"{ref_style}"
30
  src_prompt = f"{ref_prompt}, {src_style}."