debug

app.py

@@ -28,11 +28,11 @@ if not UI_ONLY_MODE:
         )
     )
 
-    # subprocess.run(
-    #     shlex.split(
-    #         "pip install ./extension/nvdiffrast-0.3.1+torch-py3-none-any.whl --force-reinstall --no-deps"
-    #     )
-    # )
+    subprocess.run(
+        shlex.split(
+            "pip install ./extension/nvdiffrast-0.3.1+torch-py3-none-any.whl --force-reinstall --no-deps"
+        )
+    )
 
     subprocess.run(
         shlex.split(

models/DiMeR/models/geometry/render/neural_render.py

@@ -69,6 +69,7 @@ def compute_vertex_normal(v_pos, t_pos_idx):
 
 
 class NeuralRender(Renderer):
+    @spaces.GPU
     def __init__(self, device='cuda', camera_model=None):
         super(NeuralRender, self).__init__()
         print(2.1)

requirements.txt

@@ -14,7 +14,6 @@ jaxtyping==0.2.36
 kiui==0.2.10
 mathutils==3.3.0
 matplotlib==3.9.2
-git+https://github.com/NVlabs/nvdiffrast
 omegaconf==2.3.0
 onnxruntime==1.18.0
 open3d==0.18.0