diff --git a/build-and-copy.sh b/build-and-copy.sh index 3526dbb..8edc5b7 100755 --- a/build-and-copy.sh +++ b/build-and-copy.sh @@ -471,6 +471,7 @@ fi COMMON_BUILD_FLAGS+=("--build-arg" "BUILD_JOBS=$BUILD_JOBS") COMMON_BUILD_FLAGS+=("--build-arg" "TORCH_CUDA_ARCH_LIST=$GPU_ARCH_LIST") COMMON_BUILD_FLAGS+=("--build-arg" "FLASHINFER_CUDA_ARCH_LIST=$GPU_ARCH_LIST") +COMMON_BUILD_FLAGS+=("--platform" "linux/arm64") if [ -n "$NETWORK_ARG" ]; then COMMON_BUILD_FLAGS+=("--network" "$NETWORK_ARG") fi