Merge pull request #5841 from shaneknapp/fix-ollama-postBuild

add postBuild to the Dockerfile
berkeley-dsep-infra · Jul 3, 2024 · 98e69a9 · 98e69a9
2 parents 9e4e3f1 + 8a74a5f
commit 98e69a9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/deployments/dev/images/default/Dockerfile b/deployments/dev/images/default/Dockerfile
@@ -125,8 +125,12 @@ RUN R -e "IRkernel::installspec(user = FALSE, prefix='${CONDA_DIR}')"
 
 # clear out /tmp
 USER root
+COPY postBuild /tmp/postBuild
+RUN chmod +x /tmp/postBuild
+RUN /tmp/postBuild
 RUN rm -rf /tmp/*
 
+
 USER ${NB_USER}
 WORKDIR /home/${NB_USER}
 

diff --git a/deployments/dev/images/default/postBuild b/deployments/dev/images/default/postBuild
@@ -13,7 +13,7 @@ wget https://github.com/ollama/ollama/releases/download/v0.1.48/ollama-linux-amd
 chmod +x /usr/local/bin/ollama
 
 # Run the phi3:mini model with the specified data directory
-/usr/local/bin/ollama run phi3:mini --data-dir $OLLAMA_DATA_DIR
+# /usr/local/bin/ollama run phi3:mini --data-dir $OLLAMA_DATA_DIR
 
 # Run the tinyllama model with the specified data directory
-/usr/local/bin/ollama run tinyllama --data-dir $OLLAMA_DATA_DIR
+# /usr/local/bin/ollama run tinyllama --data-dir $OLLAMA_DATA_DIR