https://github.com/ai-dock/comfyui
ComfyUI docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.
https://github.com/ai-dock/comfyui
ai comfyui docker image-processing runpod runpod-serverless runpod-worker sdxl stable-diffusion vast
Last synced: 5 months ago
JSON representation
ComfyUI docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.
- Host: GitHub
- URL: https://github.com/ai-dock/comfyui
- Owner: ai-dock
- License: other
- Created: 2023-08-22T19:57:50.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-11-04T18:23:58.000Z (11 months ago)
- Last Synced: 2025-05-08T03:54:39.231Z (5 months ago)
- Topics: ai, comfyui, docker, image-processing, runpod, runpod-serverless, runpod-worker, sdxl, stable-diffusion, vast
- Language: Shell
- Homepage:
- Size: 782 KB
- Stars: 840
- Watchers: 7
- Forks: 279
- Open Issues: 47
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE.md
Awesome Lists containing this project
README
[](https://github.com/ai-dock/comfyui/actions/workflows/docker-build.yml)
# AI-Dock + ComfyUI Docker Image
Run [ComfyUI](https://github.com/comfyanonymous/ComfyUI) in a highly-configurable, cloud-first AI-Dock container.
>[!NOTE]
>These images do not bundle models or third-party configurations. You should use a [provisioning script](https://github.com/ai-dock/base-image/wiki/4.0-Running-the-Image#provisioning-script) to automatically configure your container. You can find examples, including `SD3` & `FLUX.1` setup, in `config/provisioning`.
## Documentation
All AI-Dock containers share a common base which is designed to make running on cloud services such as [vast.ai](https://link.ai-dock.org/vast.ai) as straightforward and user friendly as possible.
Common features and options are documented in the [base wiki](https://github.com/ai-dock/base-image/wiki) but any additional features unique to this image will be detailed below.
#### Version Tags
The `:latest` tag points to `:latest-cuda` and will relate to a stable and tested version. There may be more recent builds
Tags follow these patterns:
##### _CUDA_
- `:cuda-[x.x.x-base|runtime]-[ubuntu-version]`
##### _ROCm_
- `:rocm-[x.x.x-runtime]-[ubuntu-version]`
##### _CPU_
- `:cpu-[ubuntu-version]`
Browse [ghcr.io](https://github.com/ai-dock/comfyui/pkgs/container/comfyui) for an image suitable for your target environment. Alternatively, view a select range of [CUDA](https://hub.docker.com/r/aidockorg/comfyui-cuda) and [ROCm](https://hub.docker.com/r/aidockorg/comfyui-rocm) builds at DockerHub.
Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`
## Additional Environment Variables
| Variable | Description |
| ------------------------ | ----------- |
| `AUTO_UPDATE` | Update ComfyUI on startup (default `false`) |
| `CIVITAI_TOKEN` | Authenticate download requests from Civitai - Required for gated models |
| `COMFYUI_ARGS` | Startup arguments. eg. `--gpu-only --highvram` |
| `COMFYUI_PORT_HOST` | ComfyUI interface port (default `8188`) |
| `COMFYUI_REF` | Git reference for auto update. Accepts branch, tag or commit hash. Default: latest release |
| `COMFYUI_URL` | Override `$DIRECT_ADDRESS:port` with URL for ComfyUI |
| `HF_TOKEN` | Authenticate download requests from HuggingFace - Required for gated models (SD3, FLUX, etc.) |
See the base environment variables [here](https://github.com/ai-dock/base-image/wiki/2.0-Environment-Variables) for more configuration options.
### Additional Python Environments
| Environment | Packages |
| -------------- | ----------------------------------------- |
| `comfyui` | ComfyUI and dependencies |
| `api` | ComfyUI API wrapper and dependencies |
The `comfyui` environment will be activated on shell login.
~~See the base micromamba environments [here](https://github.com/ai-dock/base-image/wiki/1.0-Included-Software#installed-micromamba-environments).~~
## Additional Services
The following services will be launched alongside the [default services](https://github.com/ai-dock/base-image/wiki/1.0-Included-Software) provided by the base image.
### ComfyUI
The service will launch on port `8188` unless you have specified an override with `COMFYUI_PORT_HOST`.
You can set startup flags by using variable `COMFYUI_ARGS`.
To manage this service you can use `supervisorctl [start|stop|restart] comfyui`.
### ComfyUI API Wrapper
This service is available on port `8188` and is a work-in-progress to replace previous serverless handlers which have been depreciated; Old Docker images and sources remain available should you need them.
You can access the api directly at `/ai-dock/api/` or you can use the Swager/openAPI playground at `/ai-dock/api/docs`.
>[!NOTE]
>All services are password protected by default. See the [security](https://github.com/ai-dock/base-image/wiki#security) and [environment variables](https://github.com/ai-dock/base-image/wiki/2.0-Environment-Variables) documentation for more information.
## Pre-Configured Templates
**Vast.ai**
- [comfyui:latest-cuda](https://link.ai-dock.org/template-vast-comfyui)
- [comfyui:latest-cuda + FLUX.1](https://link.ai-dock.org/template-vast-comfyui-flux)
- [comfyui:latest-rocm](https://link.ai-dock.org/template-vast-comfyui-rocm)
---
_The author ([@robballantyne](https://github.com/robballantyne)) may be compensated if you sign up to services linked in this document. Testing multiple variants of GPU images in many different environments is both costly and time-consuming; This helps to offset costs_