IOPaint/build_docker.sh

32 lines
1.0 KiB
Bash
Raw Normal View History

2022-10-08 06:55:51 +02:00
#!/usr/bin/env bash
set -e
2022-11-01 02:09:43 +01:00
GIT_TAG=$1
2022-10-08 06:55:51 +02:00
IMAGE_DESC="Image inpainting tool powered by SOTA AI Model"
GIT_REPO="https://github.com/Sanster/lama-cleaner"
echo "Building cpu docker image..."
docker buildx build \
--file ./docker/CPUDockerfile \
--label org.opencontainers.image.title=lama-cleaner \
--label org.opencontainers.image.description="$IMAGE_DESC" \
--label org.opencontainers.image.url=$GIT_REPO \
--label org.opencontainers.image.source=$GIT_REPO \
--label org.opencontainers.image.version=$GIT_TAG \
2022-11-01 02:09:43 +01:00
--build-arg version=$GIT_TAG \
2023-01-20 02:30:14 +01:00
--tag cwq1913/lama-cleaner:cpu-$GIT_TAG .
2022-10-08 06:55:51 +02:00
echo "Building NVIDIA GPU docker image..."
docker buildx build \
--file ./docker/GPUDockerfile \
--label org.opencontainers.image.title=lama-cleaner \
--label org.opencontainers.image.description="$IMAGE_DESC" \
--label org.opencontainers.image.url=$GIT_REPO \
--label org.opencontainers.image.source=$GIT_REPO \
--label org.opencontainers.image.version=$GIT_TAG \
2022-11-01 02:09:43 +01:00
--build-arg version=$GIT_TAG \
2023-01-20 02:30:14 +01:00
--tag cwq1913/lama-cleaner:gpu-$GIT_TAG .