Spaces:
Running
on
Zero
Running
on
Zero
lower zerogpu duration requirement + display tqdm tracking
Browse files
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=
|
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}."
|