Nekochu commited on
Commit
2d1916c
·
1 Parent(s): f72a3ff

Use git install for torchmcubes (ABI compatible)

Browse files
Files changed (2) hide show
  1. app.py +0 -8
  2. requirements.txt +2 -2
app.py CHANGED
@@ -6,14 +6,6 @@ Tab 2: Dust3R (Multi-view -> Point Cloud)
6
  import os
7
  import sys
8
  import subprocess
9
-
10
- # Install torchmcubes from local wheel before importing tsr
11
- wheel_path = os.path.join(os.path.dirname(__file__), "torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl")
12
- if os.path.exists(wheel_path):
13
- print(f"Installing torchmcubes from {wheel_path}...", flush=True)
14
- subprocess.run([sys.executable, "-m", "pip", "install", wheel_path, "-q"], check=True)
15
- print("torchmcubes installed!", flush=True)
16
-
17
  import logging
18
  import tempfile
19
  import time
 
6
  import os
7
  import sys
8
  import subprocess
 
 
 
 
 
 
 
 
9
  import logging
10
  import tempfile
11
  import time
requirements.txt CHANGED
@@ -1,10 +1,10 @@
1
  --extra-index-url https://download.pytorch.org/whl/cpu
2
- torch
3
  # TripoSR (AleenDG versions)
4
  omegaconf==2.3.0
5
  Pillow==10.1.0
6
  einops==0.7.0
7
- # torchmcubes installed in app.py from local wheel
8
  transformers>=4.35.0
9
  trimesh==4.0.5
10
  numpy<2.0
 
1
  --extra-index-url https://download.pytorch.org/whl/cpu
2
+ torch==2.2.0
3
  # TripoSR (AleenDG versions)
4
  omegaconf==2.3.0
5
  Pillow==10.1.0
6
  einops==0.7.0
7
+ git+https://github.com/cocktailpeanut/torchmcubes.git
8
  transformers>=4.35.0
9
  trimesh==4.0.5
10
  numpy<2.0