okaris commited on
Commit
8fd7005
β€’
1 Parent(s): 50c6839

Add accelerate and increase zero gepu duration

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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