Spaces:
Paused
Paused
Enable ZeroGPU queue (beta feature) (#2)
Browse files- Enable ZeroGPU queue (beta feature) (f284d3f7ab53318493296f9a0310b65b33040b77)
Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>
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
|