This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user