Spaces:
Paused
Paused
Fix: install diso at runtime after torch is available
Browse files- app.py +2 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -16,6 +16,8 @@ import shutil
|
|
| 16 |
subprocess.run("pip install spandrel==0.4.1 --no-deps", shell=True, check=True)
|
| 17 |
# Install nvdiffrast with --no-build-isolation for CUDA compilation
|
| 18 |
subprocess.run("pip install git+https://github.com/NVlabs/nvdiffrast.git --no-build-isolation", shell=True, check=True)
|
|
|
|
|
|
|
| 19 |
|
| 20 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 21 |
DTYPE = torch.float16
|
|
|
|
| 16 |
subprocess.run("pip install spandrel==0.4.1 --no-deps", shell=True, check=True)
|
| 17 |
# Install nvdiffrast with --no-build-isolation for CUDA compilation
|
| 18 |
subprocess.run("pip install git+https://github.com/NVlabs/nvdiffrast.git --no-build-isolation", shell=True, check=True)
|
| 19 |
+
# Install diso (needs torch to be available)
|
| 20 |
+
subprocess.run("pip install diso --no-build-isolation", shell=True, check=True)
|
| 21 |
|
| 22 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 23 |
DTYPE = torch.float16
|
requirements.txt
CHANGED
|
@@ -27,4 +27,4 @@ wheel
|
|
| 27 |
# Other dependencies
|
| 28 |
cvcuda_cu12
|
| 29 |
gltflib
|
| 30 |
-
diso
|
|
|
|
| 27 |
# Other dependencies
|
| 28 |
cvcuda_cu12
|
| 29 |
gltflib
|
| 30 |
+
# diso will be installed at runtime after torch is available
|