multimodalart HF Staff commited on
Commit
ef13f4e
·
verified ·
1 Parent(s): 85d0b1a

[Admin maintenance] Support new ZeroGPU hardware

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -14,11 +14,14 @@ from tqdm import tqdm
14
  from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
15
 
16
  import subprocess, sys
17
- subprocess.check_call([
18
- sys.executable, "-m", "pip", "install", "--no-build-isolation",
19
- "git+https://github.com/NVlabs/nvdiffrast/",
20
- ])
21
 
 
 
 
 
 
 
 
22
  from src.utils.train_util import instantiate_from_config
23
  from src.utils.camera_util import (
24
  FOV_to_intrinsics,
 
14
  from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
15
 
16
  import subprocess, sys
 
 
 
 
17
 
18
+ @spaces.GPU(duration=210)
19
+ def first_gpu_setup():
20
+ subprocess.check_call([
21
+ sys.executable, "-m", "pip", "install", "--no-build-isolation",
22
+ "git+https://github.com/NVlabs/nvdiffrast/",
23
+ ])
24
+ first_gpu_setup()
25
  from src.utils.train_util import instantiate_from_config
26
  from src.utils.camera_util import (
27
  FOV_to_intrinsics,