update

app.py

@@ -143,9 +143,10 @@ def image_to_3d(
         }[resolution],
         return_latent=True,
     )
-    print('DONE')
+    mesh = outputs[0]
+    mesh.simplify(16777216) # nvdiffrast limit
     images = render_utils.make_pbr_vis_frames(
-        render_utils.render_snapshot(outputs[0], resolution=1024, r=2, fov=36, envmap=envmap),
+        render_utils.render_snapshot(mesh, resolution=1024, r=2, fov=36, envmap=envmap),
         resolution=1024
     )
     state = pack_state(latents)

packages.txt

@@ -1,4 +0,0 @@
-libegl1
-libegl1-mesa
-libgl1-mesa-glx
-libgles2

requirements.txt

@@ -19,5 +19,5 @@ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn
 https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/cumesh-0.0.1-cp310-cp310-linux_x86_64.whl
 https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/flex_gemm-0.0.1-cp310-cp310-linux_x86_64.whl
 https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/o_voxel-0.0.1-cp310-cp310-linux_x86_64.whl
-https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/nvdiffrast-0.3.5-cp310-cp310-linux_x86_64.whl
+https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/nvdiffrast-0.4.0-cp310-cp310-linux_x86_64.whl
 https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/nvdiffrec_render-0.0.0-cp310-cp310-linux_x86_64.whl

trellis2/renderers/mesh_renderer.py

@@ -39,7 +39,6 @@ class MeshRenderer:
 
     Args:
         rendering_options (dict): Rendering options.
-        glctx (nvdiffrast.torch.RasterizeGLContext): RasterizeGLContext object for CUDA/OpenGL interop.
         """
     def __init__(self, rendering_options={}, device='cuda'):
         if 'dr' not in globals():

trellis2/renderers/pbr_mesh_renderer.py

@@ -122,7 +122,6 @@ class PbrMeshRenderer:
 
     Args:
         rendering_options (dict): Rendering options.
-        glctx (nvdiffrast.torch.RasterizeGLContext): RasterizeGLContext object for CUDA/OpenGL interop.
         """
     def __init__(self, rendering_options={}, device='cuda'):
         if 'dr' not in globals():
@@ -136,7 +135,7 @@ class PbrMeshRenderer:
             "peel_layers": 8,
         })
         self.rendering_options.update(rendering_options)
-        self.glctx = dr.RasterizeGLContext(device=device)
+        self.glctx = dr.RasterizeCudaContext(device=device)
         self.device=device
         
     def render(