Spaces:
Paused
Paused
[Admin maintenance] Support new ZeroGPU hardware
Browse files
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,
|