comfyui

comfyui

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

Stars: 434

Visit
 screenshot

ComfyUI is a highly-configurable, cloud-first AI-Dock container that allows users to run ComfyUI without bundled models or third-party configurations. Users can configure the container using provisioning scripts. The Docker image supports NVIDIA CUDA, AMD ROCm, and CPU platforms, with version tags for different configurations. Additional environment variables and Python environments are provided for customization. ComfyUI service runs on port 8188 and can be managed using supervisorctl. The tool also includes an API wrapper service and pre-configured templates for Vast.ai. The author may receive compensation for services linked in the documentation.

README:

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 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_BRANCH ComfyUI branch/commit hash for auto update (default master)
COMFYUI_ARGS Startup flags. eg. --gpu-only --highvram
COMFYUI_PORT_HOST ComfyUI interface port (default 8188)
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

For Tasks:

Click tags to check more tools for each tasks

For Jobs:

Alternative AI tools for comfyui

Similar Open Source Tools

For similar tasks

For similar jobs