This commit is contained in:
@@ -471,6 +471,7 @@ fi
|
|||||||
COMMON_BUILD_FLAGS+=("--build-arg" "BUILD_JOBS=$BUILD_JOBS")
|
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" "TORCH_CUDA_ARCH_LIST=$GPU_ARCH_LIST")
|
||||||
COMMON_BUILD_FLAGS+=("--build-arg" "FLASHINFER_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
|
if [ -n "$NETWORK_ARG" ]; then
|
||||||
COMMON_BUILD_FLAGS+=("--network" "$NETWORK_ARG")
|
COMMON_BUILD_FLAGS+=("--network" "$NETWORK_ARG")
|
||||||
fi
|
fi
|
||||||
|
|||||||
Reference in New Issue
Block a user