Update docker-compose.yml

This commit is contained in:
Stefano Azevedo 2023-01-15 12:54:39 -03:00 committed by GitHub
parent 13dfd4eba5
commit 812a93a0ff
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -3,6 +3,7 @@ version: '3.9'
x-base_service: &base_service x-base_service: &base_service
ports: ports:
- "7860:7860" - "7860:7860"
- "7861:7861"
volumes: volumes:
- &v1 ./data:/data - &v1 ./data:/data
- &v2 ./output:/output - &v2 ./output:/output
@ -11,7 +12,7 @@ x-base_service: &base_service
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
device_ids: ['0'] device_ids: ['0,1']
capabilities: [gpu] capabilities: [gpu]
name: webui-docker name: webui-docker
@ -23,13 +24,21 @@ services:
volumes: volumes:
- *v1 - *v1
auto: &automatic auto0: &automatic0
<<: *base_service <<: *base_service
profiles: ["auto"] profiles: ["auto0"]
build: ./services/AUTOMATIC1111 build: ./services/AUTOMATIC1111
image: sd-auto:33 image: sd-auto0:33
environment: environment:
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api - CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api
auto1: &automatic1
<<: *base_service
profiles: ["auto1"]
build: ./services/AUTOMATIC1111
image: sd-auto1:33
environment:
- CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api
auto-cpu: auto-cpu:
<<: *automatic <<: *automatic