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 |
||
---|---|---|
.. | ||
.env.sample | ||
build.sh | ||
docker-compose.yml | ||
docker-entrypoint.sh | ||
Dockerfile | ||
env.sh | ||
README.md | ||
run.sh |
InvokeAI Containerized
All commands are to be run from the docker
directory: cd docker
Linux
- Ensure builkit is enabled in the Docker daemon settings (
/etc/docker/daemon.json
) - Install
docker-compose
- Ensure docker daemon is able to access the GPU.
macOS
- Ensure Docker has at least 16GB RAM
- Enable VirtioFS for file sharing
- Enable
docker-compose
V2 support
This is done via Docker Desktop preferences
Quickstart
- Make a copy of
env.sample
and name it.env
(cp env.sample .env
(Mac/Linux) orcopy example.env .env
(Windows)). Make changes as necessary. docker-compose up
The image will be built automatically if needed.
The runtime directory (holding models and outputs) will be created in your home directory, under ~/invokeai
, populated with necessary content (you will be asked a couple of questions during setup)
Use a GPU
- Linux is recommended for GPU support in Docker.
- WSL2 is required for Windows.
- only
x86_64
architecture is supported.
The Docker daemon on the system must be already set up to use the GPU. In case of Linux, this involves installing nvidia-docker-runtime
and configuring the nvidia
runtime as default. Steps will be different for AMD. Please see Docker documentation for the most up-to-date instructions for using your GPU with Docker.
Customize
Check the .env
file. It contains environment variables for running in Docker. Fill it in with your own values. Next time you run docker-compose up
, your custom values will be used.
You can also set these values in docker-compose.yml
directly, but .env
will help avoid conflicts when code is updated.
Example:
LOCAL_ROOT_DIR=/Volumes/HugeDrive/invokeai
HUGGINGFACE_TOKEN=the_actual_token
CONTAINER_UID=1000
GPU_DRIVER=cuda
Moar Customize!
See the docker-compose.yaml
file. The command
instruction can be uncommented and used to run arbitrary startup commands. Some examples below.
Turn off the NSFW checker
command:
- invokeai
- --no-nsfw_check
- --web
- --host 0.0.0.0
Reconfigure the runtime directory
Can be used to download additional models from the supported model list
In conjunction with LOCAL_ROOT_DIR
can be also used to create bran
command:
- invokeai-configure
- --yes
Run in CLI mode
This container starts InvokeAI in web mode by default.
Override the command
and run `docker compose:
command:
- invoke
Then attach to the container from another terminal:
$ docker attach $(docker compose ps invokeai -q)
invoke>
Enjoy using the invoke>
prompt. To detach from the container, type Ctrl+P
followed by Ctrl+Q
(this is the escape sequence).