fix: update Dockerfile

Cinnamon · Jan 20, 2025 · bfa1b17 · bfa1b17
1 parent 0c322e6
commit bfa1b17
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -97,6 +97,9 @@ RUN --mount=type=ssh  \
     --mount=type=cache,target=/root/.cache/pip  \
     curl -fsSL https://ollama.com/install.sh | sh
 
+RUN nohup bash -c "ollama serve &" && sleep 4 && ollama pull llama3.1:8b
+RUN nohup bash -c "ollama serve &" && sleep 4 && ollama pull nomic-embed-text
+
 # Clean up
 RUN apt-get autoremove \
     && apt-get clean \