Try to avoid cudaMalloc

requirements.txt

@@ -4,9 +4,6 @@
 torch==2.4.0
 torchvision==0.19.0
 
-https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
-https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
-
 gradio
 matplotlib
 tqdm
@@ -14,6 +11,10 @@ opencv-python
 scipy
 einops
 trimesh
+
+https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
+https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
+
 tensorboard
 pyglet<2
 roma

wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d5d186b907917fba676b89dbe35e644fa617cf73781b94da0740648c09829611
-size 3089436
+oid sha256:d4c8abe48b59dad3570ea1e4d5540303f8a16869d5a61a2737b043142c220497
+size 3082355