persistent wheels

.gitattributes

@@ -34,4 +34,3 @@ 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
 *.ply filter=lfs diff=lfs merge=lfs -text
-*.whl filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -41,8 +41,8 @@ RUN pip install wheel
 RUN pip install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 torchtext==0.16.0 torchdata==0.7.0 --extra-index-url https://download.pytorch.org/whl/cu121 -U
 RUN sed -i 's/return caster.operator typename make_caster<T>::template cast_op_type<T>();/return caster;/' /home/user/.local/lib/python3.10/site-packages/torch/include/pybind11/cast.h
 RUN pip install tyro kiui PyMCubes nerfacc trimesh pymeshlab ninja plyfile xatlas pygltflib gradio opencv-python scikit-learn
-RUN pip install https://github.com/camenduru/LGM-replicate/releases/download/replicate/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
-RUN pip install https://github.com/camenduru/wheels/releases/download/colab/nvdiffrast-0.3.1-py3-none-any.whl
+RUN pip install https://github.com/dylanebert/wheels/releases/download/1.0.0/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+RUN pip install https://github.com/dylanebert/wheels/releases/download/1.0.0/nvdiffrast-0.3.1-py3-none-any.whl
 RUN pip install git+https://github.com/ashawkey/kiuikit.git
 
 # Copy all files to the working directory

convert.py

@@ -422,7 +422,7 @@ class Converter(nn.Module):
         vertices, triangles = clean_mesh(vertices, triangles, remesh=False)
 
         rotation_matrix = np.array(
-            [[1, 0, 0], [0, -1, 0], [0, 0, -1]], dtype=np.float32
+            [[-1, 0, 0], [0, -1, 0], [0, 0, 1]], dtype=np.float32
         )
         vertices = vertices @ rotation_matrix.T