From 36da9a600ec5262524197a770fb4b865b204d4f3 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Mon, 11 Nov 2024 19:30:15 -0800 Subject: [PATCH] add explicit platform --- llama_stack/distribution/build_container.sh | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/llama_stack/distribution/build_container.sh b/llama_stack/distribution/build_container.sh index 81d8811e5..59b19779e 100755 --- a/llama_stack/distribution/build_container.sh +++ b/llama_stack/distribution/build_container.sh @@ -147,8 +147,19 @@ else image_name="$image_name-$(curl -s $URL | jq -r '.info.version')" fi +# Detect platform architecture +ARCH=$(uname -m) +if [ "$ARCH" = "arm64" ] || [ "$ARCH" = "aarch64" ]; then + PLATFORM="--platform linux/arm64" +elif [ "$ARCH" = "x86_64" ]; then + PLATFORM="--platform linux/amd64" +else + echo "Unsupported architecture: $ARCH" + exit 1 +fi + set -x -$DOCKER_BINARY build $DOCKER_OPTS -t $image_name -f "$TEMP_DIR/Dockerfile" "$REPO_DIR" $mounts +$DOCKER_BINARY build $DOCKER_OPTS $PLATFORM -t $image_name -f "$TEMP_DIR/Dockerfile" "$REPO_DIR" $mounts # clean up tmp/configs set +x