Spaces:

jasperai
/

flash-sd3

Running on Zero

clementchadebec commited on 13 days ago

Commit

38047c4

•

1 Parent(s): 592c899

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ NUM_INFERENCE_STEPS = 4
 @spaces.GPU
-def infer(prompt, seed, randomize_seed, guidance_scale, num_inference_steps, negative_prompt):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)

 @spaces.GPU
+def infer(prompt, seed, randomize_seed, guidance_scale, num_inference_steps, negative_prompt, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)