Spaces:

ovi054
/

Wan2.1-Image

Running on Zero

ovi054 commited on Mar 15

Commit

2e5b3cf

verified ·

1 Parent(s): 69a9a62

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow
 @spaces.GPU()
-def generate(prompt):
     pipe.to("cuda")
     output = pipe(
         prompt=prompt,

 @spaces.GPU()
+def generate(prompt, progress=gr.Progress(track_tqdm=True)):
     pipe.to("cuda")
     output = pipe(
         prompt=prompt,