Enable ZeroGPU queue (beta feature)

#2
by cbensimon HF staff - opened
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