Hack: download go from aliyun/use goproxy.cn/only build musa image (#3)

Signed-off-by: Xiaodong Ye <[email protected]>

Author: yeahdongcn

Dockerfile

@@ -81,8 +81,9 @@ RUN --mount=type=cache,target=/root/.ccache \
 FROM base AS build
 WORKDIR /go/src/github.com/ollama/ollama
 COPY go.mod go.sum .
-RUN curl -fsSL https://golang.org/dl/go$(awk '/^go/ { print $2 }' go.mod).linux-$(case $(uname -m) in x86_64) echo amd64 ;; aarch64) echo arm64 ;; esac).tar.gz | tar xz -C /usr/local
+RUN curl -fsSL https://mirrors.aliyun.com/golang/go$(awk '/^go/ { print $2 }' go.mod).linux-$(case $(uname -m) in x86_64) echo amd64 ;; aarch64) echo arm64 ;; esac).tar.gz | tar xz -C /usr/local
 ENV PATH=/usr/local/go/bin:$PATH
+ENV GOPROXY=https://goproxy.cn,direct
 RUN go mod download
 COPY . .
 ARG GOFLAGS="'-ldflags=-w -s'"

scripts/build_docker.sh

@@ -15,16 +15,21 @@ else
     LOAD_OR_PUSH="--push"
 fi
 
-docker buildx build \
-    ${LOAD_OR_PUSH} \
-    --platform=${PLATFORM} \
-    ${OLLAMA_COMMON_BUILD_ARGS} \
-    -f Dockerfile \
-    -t ${FINAL_IMAGE_REPO}:$VERSION \
-    .
+FLAVORS="musa"
+if [ "${DOCKER_ORG}" != "mthreads" ]; then
+    docker buildx build \
+        ${LOAD_OR_PUSH} \
+        --platform=${PLATFORM} \
+        ${OLLAMA_COMMON_BUILD_ARGS} \
+        -f Dockerfile \
+        -t ${FINAL_IMAGE_REPO}:$VERSION \
+        .
+    FLAVORS="rocm musa"
+fi
+
 
 if echo $PLATFORM | grep "amd64" > /dev/null; then
-    for FLAVOR in "rocm" "musa"; do
+    for FLAVOR in $FLAVORS; do
         docker buildx build \
             ${LOAD_OR_PUSH} \
             --platform=linux/amd64 \