efecelik commited on
Commit
cf28b42
·
1 Parent(s): 1f42bcf

Fix: install diso at runtime after torch is available

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