Update Dockerfile

Dockerfile

@@ -1,28 +1,52 @@
-FROM nvidia/cuda:11.6.0-base-ubuntu20.04
-RUN export PATH="/usr/local/cuda/bin:$PATH"
+FROM nvidia/cuda:11.0.3-base-ubuntu20.04
+
+ENV MODEL_NAME="robin-13B-v2-GGML"
+ENV DEFAULT_MODEL_FILE="robin-13b.ggmlv3.q4_K_M.bin"
+ENV MODEL_USER="TheBloke"
+ENV DEFAULT_MODEL_BRANCH="main"
+ENV MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_MODEL_FILE}"
+ENV PATH="/usr/local/cuda/bin:$PATH"
+
 RUN apt update && \
     apt install --no-install-recommends -y build-essential python3 python3-pip wget curl git && \
     apt clean && rm -rf /var/lib/apt/lists/*
 
-EXPOSE 8000
-
 WORKDIR /app
 
-RUN export PATH="/usr/local/cuda/bin:$PATH"
-RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
+RUN apt-get install -y wget && \
+    wget -qO- "https://cmake.org/files/v3.18/cmake-3.18.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
+
 RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install git+https://github.com/abetlen/llama-cpp-python --no-cache-dir
 
 COPY requirements.txt ./
-RUN pip install --upgrade pip && \
-    pip install -r requirements.txt
 
-#RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash
-#RUN apt-get install git-lfs -y
-#RUN git clone https://huggingface.co/TheBloke/robin-13B-v2-GGML
-#RUN mv robin-13B-v2-GGML/robin-13b.ggmlv3.q3_K_M.bin .
-#RUN rm -rf robin-13B-v2-GGML/
+RUN pip3 install --upgrade pip && \
+    pip3 install -r requirements.txt
+
+WORKDIR /app
+
+RUN echo ${MODEL_URL} && \
+    wget -O /app/${DEFAULT_MODEL_FILE} ${MODEL_URL}
+
+RUN useradd -m -u 1000 user
+
+RUN mkdir -p /home/user/app && \
+    mv /app/${DEFAULT_MODEL_FILE} /home/user/app
+
+RUN chown -R user:user /home/user/app
+
+USER user
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH \
+    MODEL_NAME=${MODEL_NAME} \
+    MODEL_FILE=/home/user/app/${DEFAULT_MODEL_FILE} 
+
+WORKDIR $HOME/app
+
+COPY --chown=user . .
 
-COPY . .
 RUN ls -al
 
+EXPOSE 8000
+
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]