Spaces:
Runtime error
Runtime error
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
|