jammmmm commited on
Commit
a399d55
·
1 Parent(s): 8b73b24
Files changed (3) hide show
  1. gradio_app.py +2 -3
  2. requirements.txt +2 -0
  3. texture_baker/setup.py +0 -3
gradio_app.py CHANGED
@@ -1,7 +1,4 @@
1
  import os
2
-
3
- os.system("pip install ./texture_baker/ ./uv_unwrapper/")
4
-
5
  import random
6
  import tempfile
7
  import time
@@ -18,6 +15,8 @@ from gradio_pointcloudeditor import PointCloudEditor
18
  from PIL import Image
19
  from transparent_background import Remover
20
 
 
 
21
  import spar3d.utils as spar3d_utils
22
  from spar3d.models.mesh import QUAD_REMESH_AVAILABLE, TRIANGLE_REMESH_AVAILABLE
23
  from spar3d.system import SPAR3D
 
1
  import os
 
 
 
2
  import random
3
  import tempfile
4
  import time
 
15
  from PIL import Image
16
  from transparent_background import Remover
17
 
18
+ os.system("USE_CUDA=1 pip install -vv --no-build-isolation ./texture_baker ./uv_unwrapper")
19
+
20
  import spar3d.utils as spar3d_utils
21
  from spar3d.models.mesh import QUAD_REMESH_AVAILABLE, TRIANGLE_REMESH_AVAILABLE
22
  from spar3d.system import SPAR3D
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  torch==2.1.2
2
  torchvision==0.16.2
3
  einops==0.7.0
 
1
+ wheel
2
+ setuptools==69.5.1
3
  torch==2.1.2
4
  torchvision==0.16.2
5
  einops==0.7.0
texture_baker/setup.py CHANGED
@@ -52,9 +52,6 @@ def get_extensions():
52
  extra_compile_args["nvcc"].append("-g")
53
  extra_link_args.extend(["-O0", "-g"])
54
 
55
- if use_cuda:
56
- extra_compile_args["nvcc"].append("--gpu-architecture=sm_80")
57
-
58
  define_macros = []
59
  extensions = []
60
  libraries = []
 
52
  extra_compile_args["nvcc"].append("-g")
53
  extra_link_args.extend(["-O0", "-g"])
54
 
 
 
 
55
  define_macros = []
56
  extensions = []
57
  libraries = []