mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
55 lines
1.4 KiB
YAML
55 lines
1.4 KiB
YAML
# Copyright (c) 2023 Eugene Brodsky https://github.com/ebr
|
|
|
|
x-invokeai: &invokeai
|
|
image: "local/invokeai:latest"
|
|
build:
|
|
context: ..
|
|
dockerfile: docker/Dockerfile
|
|
|
|
# Create a .env file in the same directory as this docker-compose.yml file
|
|
# and populate it with environment variables. See .env.sample
|
|
env_file:
|
|
- .env
|
|
|
|
# variables without a default will automatically inherit from the host environment
|
|
environment:
|
|
# if set, CONTAINER_INVOKEAI_ROOT will override the Invoke runtime directory location *inside* the container
|
|
- INVOKEAI_ROOT=${CONTAINER_INVOKEAI_ROOT:-/invokeai}
|
|
- HF_HOME
|
|
ports:
|
|
- "${INVOKEAI_PORT:-9090}:${INVOKEAI_PORT:-9090}"
|
|
volumes:
|
|
- type: bind
|
|
source: ${HOST_INVOKEAI_ROOT:-${INVOKEAI_ROOT:-~/invokeai}}
|
|
target: ${CONTAINER_INVOKEAI_ROOT:-/invokeai}
|
|
bind:
|
|
create_host_path: true
|
|
- ${HF_HOME:-~/.cache/huggingface}:${HF_HOME:-/invokeai/.cache/huggingface}
|
|
tty: true
|
|
stdin_open: true
|
|
|
|
|
|
services:
|
|
invokeai-cuda:
|
|
<<: *invokeai
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
|
|
invokeai-cpu:
|
|
<<: *invokeai
|
|
profiles:
|
|
- cpu
|
|
|
|
invokeai-rocm:
|
|
<<: *invokeai
|
|
devices:
|
|
- /dev/kfd:/dev/kfd
|
|
- /dev/dri:/dev/dri
|
|
profiles:
|
|
- rocm
|