[FASTER DOCKER]

Dockerfile

@@ -2,6 +2,13 @@ FROM nvidia/cuda:12.1.1-devel-ubuntu22.04 as builder
 
 # Set environment variables
 ENV BASE_IMG=nvidia/cuda:12.1.1-devel-ubuntu22.04
+ENV DOCKER_BUILDKIT=1
+ENV WORLD_SIZE=4
+ENV ARTIFACTS_PATH=/app/artifacts
+ENV STORAGE_PATH=/app/storage
+
+# Set the working directory to the root
+WORKDIR /
 
 # Install Python 3.10 and other necessary system packages
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -12,15 +19,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.10 python3.10-dev python3.10-distutils python3-pip python3.10-venv openmpi-bin libopenmpi-dev \
     && python3.10 -m pip install --no-cache-dir --upgrade pip setuptools wheel
 
-# Set the working directory to the root
-WORKDIR /
-
-# # Set environment variables
-ENV WORLD_SIZE=4
-ENV ARTIFACTS_PATH=/app/artifacts
-ENV STORAGE_PATH=/app/storage
-
-
 # Copy the requirements.txt file into the container
 COPY requirements.txt .
 
@@ -38,4 +36,4 @@ COPY . /swarms-cloud
 EXPOSE 8000
 
 # Command to run the application
-CMD ["python3.10", "-m", "uvicorn", "cogvlm:main", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["python3.10", "-m", "uvicorn", "cogvlm:main", "--host", "0.0.0.0", "--port", "8000"]