update readme

This commit is contained in:
Qing
2022-10-08 12:55:51 +08:00
parent fd7c47bb95
commit 724b8c232d
2 changed files with 38 additions and 4 deletions

29
build_docker.sh Normal file
View File

@@ -0,0 +1,29 @@
#!/usr/bin/env bash
set -e
GIT_TAG=`git describe --tags --abbrev=0`
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 \
--tag lama-cleaner:cpu-$GIT_TAG .
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 \
--tag lama-cleaner:gpu-$GIT_TAG .