Spaces:
Running
on
Zero
Running
on
Zero
okaris
commited on
Commit
•
8fd7005
1
Parent(s):
50c6839
Add accelerate and increase zero gepu duration
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import gradio as gr
|
|
| 2 |
import spaces
|
| 3 |
from omni_zero import OmniZeroSingle
|
| 4 |
|
| 5 |
-
@spaces.GPU
|
| 6 |
def generate(
|
| 7 |
seed=42,
|
| 8 |
prompt="A person",
|
|
|
|
| 2 |
import spaces
|
| 3 |
from omni_zero import OmniZeroSingle
|
| 4 |
|
| 5 |
+
@spaces.GPU(duration=180)
|
| 6 |
def generate(
|
| 7 |
seed=42,
|
| 8 |
prompt="A person",
|
requirements.txt
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
# diffusers @ git+https://github.com/huggingface/diffusers.git@0d7c4790235ac00b4524b492bc2a680dcc5cf6b0
|
|
|
|
| 2 |
controlnet_aux==0.0.8
|
| 3 |
huggingface_hub==0.20.2
|
| 4 |
insightface==0.7.3
|
|
|
|
| 1 |
# diffusers @ git+https://github.com/huggingface/diffusers.git@0d7c4790235ac00b4524b492bc2a680dcc5cf6b0
|
| 2 |
+
accelerate
|
| 3 |
controlnet_aux==0.0.8
|
| 4 |
huggingface_hub==0.20.2
|
| 5 |
insightface==0.7.3
|