Skip to content

ComfyUI docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.

License

Notifications You must be signed in to change notification settings

ai-dock/comfyui

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Docker Build

AI-Dock + ComfyUI Docker Image

Run 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 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 as straightforward and user friendly as possible.

Common features and options are documented in the base 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 for an image suitable for your target environment. Alternatively, view a select range of CUDA and 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 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.

Additional Services

The following services will be launched alongside the default services 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 and environment variables documentation for more information.

Pre-Configured Templates

Vast.​ai


The author (@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