Spaces:
Paused
Paused
File size: 602 Bytes
3ec0fd4 25adfbc 3ec0fd4 25adfbc 3ec0fd4 7bf3a05 3ec0fd4 7bf3a05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
#!/bin/bash -e
if [ -n "$1" ]; then
CUDA_ARCH="$1"
BASE_IMAGE_BUILD_ARG="--build-arg CUDA_ARCH=$CUDA_ARCH"
else
BASE_IMAGE_BUILD_ARG=""
fi
[ -n "$VERBOSE" ] && ARGS="--progress plain"
(
cd base-image &&
docker build $ARGS $BASE_IMAGE_BUILD_ARG -t ghcr.io/collabora/whisperfusion-base:latest .
)
mkdir -p scratch-space
cp -r scripts/build-* scratch-space
docker run --gpus all --shm-size 64G -v "$PWD"/scratch-space:/root/scratch-space -w /root/scratch-space -it ghcr.io/collabora/whisperfusion-base:latest ./build-models.sh
docker build $ARGS -t ghcr.io/collabora/whisperfusion:latest .
|