Skip to content

Docker Image Version

Image inference with ComfyUI

A streamlined and automated environment for running ComfyUI with image/edit models, optimized for use on RunPod.

What to expect

These templates are intended for users who already want to run ComfyUI on RunPod and are comfortable following technical setup steps. No Linux expertise is required for normal use, but basic familiarity with RunPod pods, logs, tokens, and file management is helpful.

When to use this template

Use this template for image generation and image editing workflows with Z-Image, ERNIE-Image, FLUX.2, FLUX.2 Klein, Qwen-Image, JoyCaption and Qwen-Image-Edit models.

Example ZIB-ZIT

ZIB-ZIT

Example FLUX-Klein control/target image generation

FLUX-Klein

Example Qwen-Image-Edit multiple angles

Qwen-Image-Edit camera angles

Example FLUX.2 Dev multiple angles

FLUX2-CAMERA

Example JoyCaption

JoyCaption workflow

🔧 Features

  • Automatic model and LoRA provisioning via environment variables.
  • Included workflows for image generation and enhancement using pre-installed custom nodes.
  • Compatible with high-performance NVIDIA GPUs (CUDA 12.8).
  • Compiled attention and GPU acceleration.
  • Automatic selection of bf16 or fp8 models/workflows.
  • LoRA Manager.

🔧 Built-in authentication

  • ComfyUI
  • Code Server
  • Hugging Face API
  • CivitAI API

📦 Deployment on RunPod

📘 Tutorial

Example running Z-Image on an RTX A4500

Pod running on RTX A4500 ZIT ComfyUI

Example running Z-Image on an RTX A5000

Pod running on RTX A5000 ZIT ComfyUI

Example running Z-Image on an RTX 4000 Ada

Pod running on RTX 4000 Ada ZIT ComfyUI

Example running Z-Image on an RTX 3090

Pod running on RTX 3090 ZIT ComfyUI

Example running FLUX.2 Klein 9B on an RTX A4500

Pod running on RTX A4500 FLUX.2 Klein ComfyUI

Example running Qwen-Image-Edit fp8 on an RTX A5000

Pod running on RTX A5000 Qwen-Image-Edit fp8 ComfyUI

Example running Qwen-Image-Edit bf16 on an A40

Pod running on A40 Qwen-Image-Edit bf16 ComfyUI

Example running FLUX.2 Dev bf16 on an L40S

Pod running on L40S FLUX.2 Dev bf16 ComfyUI

Example running FLUX.2 Dev fp8 on an RTX A5000 (slow)

Pod running on RTX A5000 FLUX.2 Dev fp8 ComfyUI