Move label to x-gpu-config, add "x", use git image

This commit is contained in:
AJ Walter 2023-04-15 14:59:17 -05:00
parent a7036c6072
commit 40fd30092d
No known key found for this signature in database
GPG key ID: 35E1B7402B490F34
3 changed files with 19 additions and 16 deletions

View file

@ -7,9 +7,6 @@ x-base_service: &base_service
- &v1 ./data:/data
- &v2 ./output:/output
stop_signal: SIGINT
security_opt:
- label=type:nvidia_container_t
runtime: nvidia
deploy:
resources:
reservations:
@ -18,6 +15,11 @@ x-base_service: &base_service
device_ids: ['0']
capabilities: [gpu]
x-gpu-config: &gpu_config
security_opt:
- label=type:nvidia_container_t
runtime: nvidia
name: webui-docker
services:
@ -27,23 +29,25 @@ services:
volumes:
- *v1
auto: &automatic
auto-cpu: &automatic
<<: *base_service
profiles: ["auto"]
profiles: ["auto-cpu"]
build: ./services/AUTOMATIC1111
image: sd-auto:51
environment:
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
- CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
auto-cpu:
auto:
<<: *automatic
profiles: ["auto-cpu"]
<<: *gpu_config
profiles: ["auto"]
deploy: {}
environment:
- CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
invoke:
<<: *base_service
<<: *gpu_config
profiles: ["invoke"]
build: ./services/invoke/
image: sd-invoke:26
@ -54,6 +58,7 @@ services:
sygil: &sygil
<<: *base_service
<<: *gpu_config
profiles: ["sygil"]
build: ./services/sygil/
image: sd-sygil:16

View file

@ -1,8 +1,6 @@
# syntax=docker/dockerfile:1
FROM alpine:3.17 as download
RUN apk add git
FROM alpine/git:2.36.2 as download
COPY clone.sh /clone.sh
@ -46,7 +44,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
ENV ROOT=/stable-diffusion-webui
COPY --from=download /git/ ${ROOT}
COPY --from=download /repositories/ ${ROOT}/repositories/
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/data/* ${ROOT}/interrogate
RUN --mount=type=cache,target=/root/.cache/pip \
pip install -r ${ROOT}/repositories/CodeFormer/requirements.txt

View file

@ -1,9 +1,9 @@
#!/bin/bash
set -Eeuo pipefail
set -Eeuox pipefail
mkdir -p git/repositories/"$1"
cd git/repositories/"$1"
mkdir -p /repositories/"$1"
cd /repositories/"$1"
git init
git remote add origin "$2"
git fetch origin "$3" --depth=1