ComfyUI-Attention-Distillation
Non-native [a/AttentionDistillation](https://github.com/xugao97/AttentionDistillation) for ComfyUI. Official ComfyUI demo for the paper AttentionDistillation, implemented as an extension of ComfyUI. Note that this extension incorporates AttentionDistillation using diffusers.
How much VRAM does ComfyUI-Attention-Distillation require?
Direct Answer: The ComfyUI node ComfyUI-Attention-Distillation requires a minimum base VRAM of 128MB and is optimized for GPUs with at least 4GB of VRAM. Low VRAM mode is fully supported for resource-constrained setups.
- Base VRAM:
- 128MB (0.1GB)
- Recommended GPU:
- 4GB+ VRAM
- Low VRAM Mode:
- ✓ Supported
- Estimation Confidence:
- MEDIUM
Interactive VRAM Compatibility Estimator
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 Cloud GPUs
Need more VRAM to run ComfyUI with this node? Rent low-cost, high-performance cloud GPUs on RunPod instantly.
What Python packages are required for ComfyUI-Attention-Distillation?
Direct Answer: Running ComfyUI-Attention-Distillation requires installing the following Python package dependencies: accelerate, diffusers, pillow, protobuf, sentencepiece, torch>=2.1.0, tqdm. This node specifically requires PyTorch version 2.1.0 or newer.
accelerate
diffusers
pillow
protobuf
sentencepiece
torch>=2.1.0
tqdmInteractive Setup & Dependency Resolver
# Loading command...Special Environment Requirements:
Requires PyTorch version 2.1.0+.
# Loading PyTorch command...Frequently Asked Questions
How much VRAM does ComfyUI-Attention-Distillation require?
ComfyUI-Attention-Distillation requires a minimum of 128MB (0.1GB) of VRAM for base operation. For optimal performance, a GPU with at least 4GB of VRAM is recommended. This node supports low VRAM mode for resource-constrained setups.
Can I run ComfyUI-Attention-Distillation on an RTX 3060, RTX 4070, or RTX 4090?
✅ RTX 3060 (12GB): Yes, fully compatible with 10.7GB headroom. ✅ RTX 4070 (12GB): Yes, fully compatible with 10.7GB headroom. ✅ RTX 4070 Ti (16GB): Yes, fully compatible with 14.3GB headroom. ✅ RTX 4090 (24GB): Yes, fully compatible with 21.5GB headroom
What PyTorch version does ComfyUI-Attention-Distillation need?
ComfyUI-Attention-Distillation requires the following PyTorch-related packages: torch>=2.1.0. 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 ComfyUI-Attention-Distillation?
To run ComfyUI-Attention-Distillation, you need to install: accelerate, diffusers, pillow, protobuf, sentencepiece, torch>=2.1.0, tqdm. You can install these using pip or add them to your requirements.txt file.
How can I reduce VRAM usage when running ComfyUI-Attention-Distillation?
ComfyUI-Attention-Distillation supports low VRAM mode. To reduce memory usage: (1) Enable --lowvram or --medvram flags in ComfyUI, (2) Reduce batch size to 1, (3) Use fp16 or fp8 precision if supported, (4) Close other GPU applications.
How do I install ComfyUI-Attention-Distillation in ComfyUI?
To install ComfyUI-Attention-Distillation: (1) Navigate to your ComfyUI/custom_nodes directory, (2) Clone the repository: git clone https://github.com/zichongc/ComfyUI-Attention-Distillation, (3) Install dependencies: pip install -r requirements.txt (if present), (4) Restart ComfyUI. Alternatively, use ComfyUI Manager for one-click installation.