HCP-Diffusion V2 is built on top of the 🐱 RainbowNeko Engine. Using python format config file instead of yaml, which can be more extensible, flexible, and user-friendly.
- More user-friendly config file
- Simplified configuration
- More user-friendly model evaluation
- More model and method support
✨ Features
📦 Model Support
| Model Name |
Status |
| Stable Diffusion 1.5 |
✅ Supported |
| Stable Diffusion XL (SDXL) |
✅ Supported |
| PixArt |
✅ Supported |
| FLUX |
🚧 In Development |
| Stable Diffusion 3 (SD3) |
🚧 In Development |
🧠 Fine-Tuning Capabilities
| Feature |
Description/Support |
| LoRA Layer-wise Configuration |
✅ Supported (including Conv2d) |
| Layer-wise Fine-Tuning |
✅ Supported |
| Multi-token Prompt-Tuning |
✅ Supported |
| Layer-wise Model Merging |
✅ Supported |
| Custom Optimizers |
✅ Supported (Lion, DAdaptation, pytorch-optimizer, etc.) |
| Custom LR Schedulers |
✅ Supported |
🧩 Extension Method Support
| Method |
Status |
| ControlNet (including training) |
✅ Supported |
| DreamArtist / DreamArtist++ |
✅ Supported |
| Token Attention Adjustment |
✅ Supported |
| Max Sentence Length Extension |
✅ Supported |
| Textual Inversion (Custom Tokens) |
✅ Supported |
| CLIP Skip |
✅ Supported |
🚀 Training Acceleration
🗂 Dataset Support
| Feature |
Description |
| Aspect Ratio Bucket (ARB) |
✅ Auto-clustering supported |
| Multi-source / Multi-dataset |
✅ Supported |
| LMDB |
✅ Supported |
| webdataset |
🚧 In Development |
| Local Attention Enhancement |
✅ Supported |
| Tag Shuffling & Dropout |
✅ Multiple tag editing strategies |
📉 Supported Loss Functions
| Loss Type |
Description |
| Min-SNR |
✅ Supported |
| SSIM |
✅ Supported |
| GWLoss |
✅ Supported |
🌫 Supported Diffusion Strategies
| Strategy Type |
Status |
| DDPM |
✅ Supported |
| EDM |
✅ Supported |
| Flow Matching |
✅ Supported |
🧠 Automatic Evaluation (Step Selection Assistant)
| Feature |
Description/Status |
| Image Preview |
✅ Supported (workflow preview) |
| FID |
🚧 In Development |
| CLIP Score |
🚧 In Development |
| CCIP Score |
🚧 In Development |
| Corrupt Score |
🚧 In Development |
