mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
f3b45d0ad9
- rewrite Dockerfile - add a stage to build the UI - add docker-compose.yml - add docker-entrypoint.sh such that any command may be used at runtime - docker-compose adds .env support - add a sample .env file
47 lines
1.3 KiB
YAML
47 lines
1.3 KiB
YAML
# Copyright (c) 2023 Eugene Brodsky https://github.com/ebr
|
|
|
|
version: '3.8'
|
|
|
|
services:
|
|
invokeai:
|
|
image: "local/invokeai:latest"
|
|
# edit below to run on a container runtime other than nvidia-container-runtime.
|
|
# not yet tested with rocm/AMD GPUs
|
|
# Comment out the "deploy" section to run on CPU only
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
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}
|
|
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
|