Update Dockerfile

Dockerfile

@@ -8,12 +8,10 @@ RUN apt-get update -y && apt-get install -y wget git vim ninja-build \
 # Clone the InstantSplat repository and download the pre-trained model
 RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git && \
     cd InstantSplat && \
-    git submodule update --init --recursive
-    # git submodule update --init --recursive && \
-    # cd submodules/dust3r/ && \
-    # mkdir -p checkpoints/ && \
-    # wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/ && \
-    # cd ../../
+    git submodule update --init --recursive && \
+    cd submodules/dust3r/ && \
+    mkdir -p checkpoints/ && \
+    wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/
 
 # Set up environment variables
 ENV CUDA_HOME=/usr/local/cuda
@@ -36,20 +34,14 @@ ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 #     sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
 #     pip install submodules/diff-gaussian-rasterization
 
-RUN cd InstantSplat && \
-    pip install -r requirements.txt && \
+RUN pip install -r requirements.txt && \
+    cd InstantSplat && \
     pip install submodules/simple-knn && \
     sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
     pip install submodules/diff-gaussian-rasterization && \
     cd submodules/dust3r/croco/models/curope/ && \
     python setup.py build_ext --inplace
 
-# Clone the InstantSplat repository and download the pre-trained model
-RUN cd InstantSplat && \
-    cd submodules/dust3r/ && \
-    mkdir -p checkpoints/ && \
-    wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/ && \
-    cd ../../
 
 RUN useradd -m -u 1000 user
 USER user