From 41a3bc8202e3c0c45d82a1705e944388da067048 Mon Sep 17 00:00:00 2001 From: biswaroop1547 Date: Wed, 26 Jul 2023 22:43:06 +0000 Subject: [PATCH] add: support for building SD-XL with and without refiner images --- scripts/dfs_diffusers.sh | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/scripts/dfs_diffusers.sh b/scripts/dfs_diffusers.sh index 9caceff..2d70536 100755 --- a/scripts/dfs_diffusers.sh +++ b/scripts/dfs_diffusers.sh @@ -21,13 +21,30 @@ docker buildx build --push \ ./dfs-diffusers docker buildx build --push \ - --cache-from=ghcr.io/premai-io/diffuser-stable-diffusion-2-base-gpu:latest \ + --cache-from=ghcr.io/premai-io/diffuser-stable-diffusion-2-gpu:latest \ --file ./dfs-diffusers/docker/gpu/Dockerfile \ --build-arg="MODEL_ID=stabilityai/stable-diffusion-2" \ --tag ghcr.io/premai-io/diffuser-stable-diffusion-2-gpu:latest \ --tag ghcr.io/premai-io/diffuser-stable-diffusion-2-gpu:$VERSION \ ./dfs-diffusers +docker buildx build --push \ + --cache-from=ghcr.io/premai-io/diffuser-stable-diffusion-xl-gpu:latest \ + --file ./dfs-diffusers/docker/gpu/Dockerfile \ + --build-arg="MODEL_ID=stabilityai/stable-diffusion-xl-base-1.0" \ + --tag ghcr.io/premai-io/diffuser-stable-diffusion-xl-gpu:latest \ + --tag ghcr.io/premai-io/diffuser-stable-diffusion-xl-gpu:$VERSION \ + ./dfs-diffusers + +docker buildx build --push \ + --cache-from=ghcr.io/premai-io/diffuser-stable-diffusion-xl-with-refiner-gpu:latest \ + --file ./dfs-diffusers/docker/gpu/Dockerfile \ + --build-arg="MODEL_ID=stabilityai/stable-diffusion-xl-base-1.0" \ + --build-arg="REFINER_ID=stabilityai/stable-diffusion-xl-refiner-1.0" \ + --tag ghcr.io/premai-io/diffuser-stable-diffusion-xl-with-refiner-gpu:latest \ + --tag ghcr.io/premai-io/diffuser-stable-diffusion-xl-with-refiner-gpu:$VERSION \ + ./dfs-diffusers + # docker system prune --all --force --volumes docker buildx build --push \