update build_docker.sh

ollama · Mar 21, 2024 · fc7bffd · fc7bffd
1 parent caac56e
commit fc7bffd
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/scripts/build_docker.sh b/scripts/build_docker.sh
@@ -51,6 +51,16 @@ if [ -z "${OLLAMA_SKIP_IMAGE_BUILD}" ]; then
             -f Dockerfile \
             -t ${ARCH_IMAGE_REPO}:$VERSION-rocm \
             .
+
+        docker build \
+            --load \
+            --platform=linux/amd64 \
+            --build-arg=VERSION \
+            --build-arg=GOFLAGS \
+            --target runtime-oneapi \
+            -f Dockerfile \
+            -t ${ARCH_IMAGE_REPO}:$VERSION-oneapi \
+            .
     fi
 fi
 
@@ -75,12 +85,3 @@ if [ -z "${OLLAMA_SKIP_MANIFEST_CREATE}" ]; then
         echo "  ${ARCH_IMAGE_REPO}:$VERSION-rocm"
     fi
 fi
-docker build \
-    --load \
-    --platform=linux/amd64 \
-    --build-arg=VERSION \
-    --build-arg=GOFLAGS \
-    --target runtime-oneapi \
-    -f Dockerfile \
-    -t ollama/ollama:$VERSION-oneapi \
-    .