fffiloni cbensimon HF staff commited on
Commit
784addb
·
1 Parent(s): fed1b3d

Enable ZeroGPU queue (beta feature) (#2)

Browse files

- Enable ZeroGPU queue (beta feature) (f284d3f7ab53318493296f9a0310b65b33040b77)


Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ pipe = pipe.to("cuda")
16
  pipe.enable_model_cpu_offload()
17
  pipe.enable_vae_slicing()
18
 
19
- @spaces.GPU
20
  def infer(prompt):
21
  print(f"""
22
  —/n
 
16
  pipe.enable_model_cpu_offload()
17
  pipe.enable_vae_slicing()
18
 
19
+ @spaces.GPU(enable_queue=True)
20
  def infer(prompt):
21
  print(f"""
22
  —/n