env

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 *.png filter=lfs diff=lfs merge=lfs -text
+*.whl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,4 +1,5 @@
 import os
+import shlex
 import spaces
 import subprocess
 def install_cuda_toolkit():
@@ -16,7 +17,10 @@ def install_cuda_toolkit():
     )
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 install_cuda_toolkit()
-os.system('bash install.sh')
+print("cd step1x3d_texture/differentiable_renderer/ && bash compile_mesh_painter.sh")
+os.system("cd step1x3d_texture/differentiable_renderer/ && bash compile_mesh_painter.sh")
+print('install custom')
+subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
 import time
 import uuid
 import torch

custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1dc5bea62f7ef924b9f58722b9f7634501b05af2b9507e736c256d6b2b9d90fc
+size 4674364

install.sh

@@ -1,6 +0,0 @@
-nvcc -V
-cd step1x3d_texture/custom_rasterizer
-python setup.py install
-cd ../differentiable_renderer
-python setup.py install
-cd ../../