mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
62 lines
1.6 KiB
YAML
62 lines
1.6 KiB
YAML
# Copyright (c) 2023 Eugene Brodsky https://github.com/ebr
|
|
|
|
version: '3.8'
|
|
|
|
x-invokeai: &invokeai
|
|
image: "local/invokeai:latest"
|
|
build:
|
|
context: ..
|
|
dockerfile: docker/Dockerfile
|
|
|
|
# variables without a default will automatically inherit from the host environment
|
|
environment:
|
|
- INVOKEAI_ROOT
|
|
- HF_HOME
|
|
|
|
# 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
|
|
|
|
ports:
|
|
- "${INVOKEAI_PORT:-9090}:9090"
|
|
volumes:
|
|
- ${INVOKEAI_ROOT:-~/invokeai}:${INVOKEAI_ROOT:-/invokeai}
|
|
- ${HF_HOME:-~/.cache/huggingface}:${HF_HOME:-/invokeai/.cache/huggingface}
|
|
# - ${INVOKEAI_MODELS_DIR:-${INVOKEAI_ROOT:-/invokeai/models}}
|
|
# - ${INVOKEAI_MODELS_CONFIG_PATH:-${INVOKEAI_ROOT:-/invokeai/configs/models.yaml}}
|
|
tty: true
|
|
stdin_open: true
|
|
|
|
# # Example of running alternative commands/scripts in the container
|
|
# command:
|
|
# - bash
|
|
# - -c
|
|
# - |
|
|
# invokeai-model-install --yes --default-only --config_file ${INVOKEAI_ROOT}/config_custom.yaml
|
|
# invokeai-nodes-web --host 0.0.0.0
|
|
|
|
services:
|
|
invokeai-nvidia:
|
|
<<: *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
|