CaptionForge

By DamkohlerView on GitHub →

A model-agnostic captioning framework for ComfyUI designed to generate cleaner, more consistent, auditable captions for LoRA dataset preparation through structured comparison, repeated observation, and consensus-based refinement.

How much VRAM does CaptionForge require?

Direct Answer: The ComfyUI node CaptionForge requires a minimum base VRAM of 4096MB and is optimized for GPUs with at least 8GB of VRAM. Low VRAM mode is not supported for this node.

Very High (4-8GB)
Base VRAM:
4096MB (4.0GB)
Recommended GPU:
8GB+ VRAM
Low VRAM Mode:
✗ Not supported
Estimation Confidence:
MEDIUM

Interactive VRAM Compatibility Estimator

Estimated Total VRAM: 3.00 GBTarget: 8 GB
✅ Comfortable Fit

Your GPU has plenty of headroom. You can run this node safely with your active configurations!

Deploy on High-Performance GPUs

Need more VRAM to run ComfyUI with this node? Rent low-cost, high-performance GPUs on Vast.ai instantly.

🚀 Deploy on Vast.ai

Deploy on Cloud GPUs

Need more VRAM to run ComfyUI with this node? Rent low-cost, high-performance cloud GPUs on RunPod instantly.

🚀 Deploy on RunPod

What Python packages are required for CaptionForge?

Direct Answer: Running CaptionForge requires installing the following Python package dependencies: accelerate, huggingface-hub, numpy, pillow, qwen-vl-utils, safetensors, torch, transformers. Ensure your ComfyUI environment has these packages active before launching.

requirements.txt
accelerate
huggingface-hub
numpy
pillow
qwen-vl-utils
safetensors
torch
transformers

Interactive Setup & Dependency Resolver

Operating System:
Environment Type:
Run this terminal command in your ComfyUI root folder:
# Loading command...

Related Custom Nodes

The following custom nodes run in the same PyTorch / CUDA environment and share similar categories or workflows:

AITocha Vision
by aitocha128MB VRAM
AI_Text_Comfyui
by Zch6111256MB VRAM
AK XZ Axis (XY for any KSampler)
by akawana256MB VRAM
ArtyClaw Comfy Nodes
by artyclaw256MB VRAM

Frequently Asked Questions

How much VRAM does CaptionForge require?

CaptionForge requires a minimum of 4096MB (4.0GB) of VRAM for base operation. For optimal performance, a GPU with at least 8GB of VRAM is recommended. Low VRAM mode is not supported for this node.

Can I run CaptionForge on an RTX 3060, RTX 4070, or RTX 4090?

✅ RTX 3060 (12GB): Yes, fully compatible with 6.8GB headroom. ✅ RTX 4070 (12GB): Yes, fully compatible with 6.8GB headroom. ✅ RTX 4070 Ti (16GB): Yes, fully compatible with 10.4GB headroom. ✅ RTX 4090 (24GB): Yes, fully compatible with 17.6GB headroom

What PyTorch version does CaptionForge need?

CaptionForge requires the following PyTorch-related packages: torch. Ensure your ComfyUI environment has these installed. Ensure your PyTorch installation matches your CUDA version (use torch.version.cuda to check).

What Python packages are required for CaptionForge?

To run CaptionForge, you need to install: accelerate, huggingface-hub, numpy, pillow, qwen-vl-utils, safetensors, torch, transformers. You can install these using pip or add them to your requirements.txt file.

How do I install CaptionForge in ComfyUI?

To install CaptionForge: (1) Navigate to your ComfyUI/custom_nodes directory, (2) Clone the repository: git clone https://github.com/Damkohler/CaptionForge, (3) Install dependencies: pip install -r requirements.txt (if present), (4) Restart ComfyUI. Alternatively, use ComfyUI Manager for one-click installation.