load model, fix pipeline

smerschjohann · Feb 14, 2025 · 0202050 · 0202050
1 parent 3636ef5
commit 0202050
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/.github/workflows/build-amd64-only.yml b/.github/workflows/build-amd64-only.yml
@@ -1,4 +1,4 @@
-name: Build, Merge, and Push
+name: Build AMD64 images
 
 permissions:
   packages: write
@@ -46,7 +46,7 @@ jobs:
               continue
             fi
 
-            docker buildx build
+            docker buildx build \
               -f $dir/Dockerfile \
               -t $image_name:$TAG \
               --target $target \

diff --git a/images/ollama/Dockerfile b/images/ollama/Dockerfile
@@ -4,7 +4,9 @@ ENV OLLAMA_HOST=0.0.0.0
 
 EXPOSE 11434
 
+# load model and store in image
 RUN ollama serve & \
     serve_pid=$!; sleep 15; \
-    ollama pull aratan/DeepSeek-R1-32B-Uncensored:latest; \
-    kill $serve_pid
+    ollama pull aratan/DeepSeek-R1-32B-Uncensored:latest; sleep 5; \
+    kill $serve_pid; sleep 10; \
+    echo "download completed"