add: support for building SD-XL with and without refiner images

premAI-io · Jul 26, 2023 · 41a3bc8 · 41a3bc8
1 parent a9a7507
commit 41a3bc8
Showing 1 changed file with 18 additions and 1 deletion.
diff --git 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 \