Flux LoRA Training Tools Compared
Compare Civit Trainer, Kohya_ss, ai-toolkit, and fluxgym for training Flux LoRA models, with setup notes and source links.
On this page
- TL;DR
- Why Flux LoRA Training Has a Special Challenge
- Civit Website Trainer: The Browser-First Option
- Kohya_ss: GUI Wrapper for Kohya's Scripts
- ai-toolkit: The Open-Source All-in-One Suite
- fluxgym: Low-VRAM Web UI
- Comparison Summary
- VRAM Requirements: What's Documented
- License and Model Considerations
- Common Errors and Fixes
- FAQ
- Sources
Flux LoRA Training Tools Compared
Four tools cover the Flux LoRA training landscape: the Civit website trainer, Kohya_ss, ai-toolkit, and fluxgym. Each targets a different skill level and hardware profile.
TL;DR
- Civit Trainer: Browser-based Flux LoRA training with zero local setup.
- Kohya_ss: GUI wrapper around Kohya's terminal training scripts; full parameter control.
- ai-toolkit: Open-source YAML-driven suite recommended by its README for consumer GPUs.
- fluxgym: Low-VRAM web UI targeting Flux LoRA training on constrained hardware.
Why Flux LoRA Training Has a Special Challenge
FLUX.1-dev was distilled using a reduced guidance approach. Trainers report this makes direct LoRA training on FLUX.1-dev diverge from the original training process (Flux-Dev2Pro model card). The common workaround is a two-stage approach: recover a more capable base by training on FLUX.1-dev or Flux-Dev2Pro, then train the LoRA on that recovered model.
Civit Website Trainer: The Browser-First Option
The Civit platform hosts a built-in LoRA trainer for Flux. One user compared three runs using the same dataset of AssassinKahb-style skeleton images (article 6748).
| Run | Resolution | Optimizer | Epochs | Steps Formula | Source |
|---|---|---|---|---|---|
| V1 | 512 | adamw8bit | 5 | Image Count × Repeats × Epochs ÷ Batch Size | article 6748 |
| V2 | 1024 | adamw8bit | 5 | Same formula | article 6748 |
| V3 | 1024 | Prodigy (Kappa Neuro config) | 12 | Same formula; capped at 10,000 on Civit | article 6748 |
V3 used Kappa Neuro's Prodigy configuration, which sets network dimension to 2, learning rate to 0.0005, cosine scheduler with 3 cycles, noise offset of 0.1, and min-SNR gamma of 5 (article 6748). Each run produces a .safetensors LoRA file. Civit caps training at 10,000 steps (article 6748).
The Civit Trainer requires no local installation, making it the lowest barrier to entry.
Kohya_ss: GUI Wrapper for Kohya's Scripts
Kohya_ss is a GUI wrapper around Kohya's training scripts. The Civit setup guide lists these Windows steps (article 8073):
- Clone:
git clone https://github.com/bmaltais/kohya_ss - Enter folder, checkout master,
pip install rich,setup.bat
The setup menu offers Accelerate configuration, GPU selection, and precision mode. On RTX 30/40/50 cards, select bf16; older hardware uses fp16.
Caution: The current Kohya_ss README does not mention setup.bat or pip install rich; its installer centers on uv or pip install -r requirements.txt, and lists python >= 3.10 with 3.12 recommended rather than Python 3.10.11 as a fixed requirement. Treat article 8073's steps as one reporter's procedure, not the project's current standard.
ai-toolkit: The Open-Source All-in-One Suite
ai-toolkit is a training suite for diffusion models. It supports FLUX.1-dev natively and provides both a CLI and a web UI (ai-toolkit README).
Installation
On Linux or Windows with Python >= 3.10, create and activate a venv, then install dependencies with pip install -r requirements.txt. The README shows Python 3.12 as the recommended version (ai-toolkit README).
Training a Flux LoRA
- Copy the example config:
config/examples/train_lora_flux_24gb.yamlto theconfigfolder (ai-toolkit README). - Edit the config file for your model path, dataset directory, and output name.
- Run:
python run.py config/your_config.yml
Using Flux-Dev2Pro
Article 7809 describes swapping the transformer to Flux-Dev2Pro: replace the FLUX.1-dev/transformer directory with Flux1-dev2Pro/transformer inside your training tool's model folder (article 7809). This swap is not documented in ai-toolkit's README.
fluxgym: Low-VRAM Web UI
fluxgym is an alternative interface for low-VRAM Flux LoRA training. The repository describes it as a "dead simple FLUX LoRA training UI with LOW VRAM support" (GitHub). It targets users who want a GUI rather than Kohya's terminal workflow. Exact VRAM figures from community spreadsheets were not independently verified from opened documentation.
Comparison Summary
| Feature | Civit Trainer | Kohya_ss | ai-toolkit | fluxgym |
|---|---|---|---|---|
| Interface | Browser | GUI wrapper | CLI / Web UI | Low-VRAM Web UI |
| Local Install | None | Required | Required | Optional (Docker) |
| Config Style | Form fields | GUI controls | YAML files | Web form |
| Flux Support | Native | Native | Native | Native |
| Source | article 6748 | article 8073 | GitHub | GitHub |
For related reading on Flux model architecture, see How Flux Diffusion Models Work. For GPU selection guidance, see Choosing a GPU for Diffusion Training. For a practical training workflow walkthrough, see Flux LoRA Training Step-by-Step.
VRAM Requirements: What's Documented
Article 7619 records that one early low-VRAM Flux trainer produced a LoRA in about 12 hours, and that the author tested resolutions from 256 to 1024 before settling on practical settings (article 7619). The article itself was later updated by a newer VRAM-management guide. Exact time-reduction percentages from the community spreadsheet are omitted here because they were not independently verified.
License and Model Considerations
The Flux1-dev2pro model is distributed under the FLUX.1 [dev] Non-Commercial License (Civitai model 654919, Flux-Dev2Pro model card). LoRA models trained with it inherit this restriction for commercial deployment.
The base FLUX.1-dev model is similarly licensed by Black Forest Labs under the FLUX.1 dev Non-Commercial License (Hugging Face repo). Production or commercial use of trained models requires a separate license from Black Forest Labs.
Common Errors and Fixes
| Problem | Cause | Fix |
|---|---|---|
Kohya_ss setup.bat not found | Article 8073 used older commands; the current Kohya_ss README centers on uv or pip install -r requirements.txt and lists python >= 3.10 | Follow the Kohya_ss README for the current setup flow |
| PyTorch nightly wheel fails to install | pip install --pre torch ... nightly/cu128 depends on nightly artifact availability | Use a stable PyTorch release or verify the nightly index is serving your CUDA version |
| Transformer swap path missing in ai-toolkit | The FLUX.1-dev/transformer → Flux1-dev2Pro/transformer swap is documented in article 7809, not in ai-toolkit's README | Inspect your local model folders before assuming the path exists |
| Training stalls on a 12 GB card | Default batch size and resolution consume too much VRAM | Start at 512×512 or 768×768, lower batch size, and enable gradient accumulation |
FAQ
How do I train Flux LoRA on consumer GPUs? Use ai-toolkit with its example Flux config, or the Civit website trainer for a no-install browser workflow. On low VRAM, start at 512×512 with a small batch size.
What is the best VRAM setup for Flux training? Lower resolution and smaller batch sizes reduce VRAM linearly. Gradient accumulation lets you simulate larger batches without VRAM spikes.
How do I swap the transformer for Flux1-dev2pro? Article 7809 instructs replacing the FLUX.1-dev/transformer directory with Flux1-dev2Pro/transformer inside your training tool's model folder. This exact path is not documented in ai-toolkit's README.
What are recommended settings for Kohya_ss on Windows? Clone the repo, run the setup menu, select bf16 for RTX 30/40/50 series cards, disable DeepSpeed and distributed training, and enable NUMA for NVIDIA hardware. The current Kohya_ss README uses uv or pip install -r requirements.txt rather than setup.bat and pip install rich.
Is Flux1-dev2pro free to use? The model is publicly available under the FLUX.1 [dev] Non-Commercial License. You may use it for research and personal projects but must obtain a commercial license from Black Forest Labs for production deployment.
What is the difference between Kohya_ss and fluxgym? Kohya_ss is a GUI wrapper around Kohya's training scripts. fluxgym is a separate low-VRAM web UI that targets Flux LoRA training on constrained hardware.
Sources
| Source | Contribution |
|---|---|
| Kohya_ss GitHub repository | Tool installation and GUI setup guide |
| ai-toolkit GitHub repository | Open-source training suite containing CLI and web UI for Flux LoRA |
| fluxgym GitHub repository | Low-VRAM web UI targeting Flux LoRA training |
| Flux-Dev2Pro Hugging Face model card | Source of the Flux1-dev2pro transformer used for LoRA training |
| Civitai article 6748 — Training FLUX Lora with Civit | Community comparison of Civit model training runs and configuration details |
| Civitai article 8073 — How to install Kohya_SS + PyTorch 2.7.0 | Windows setup instructions for Kohya_ss and PyTorch installation |
| Civitai article 7809 — How to Train LoRA Using Flux1-dev2pro with ai-toolkit | ai-toolkit integration guide for Flux1-dev2pro LoRA training |
| Civitai article 7619 — Training Flux: Optimal Settings for a RTX 3060 | VRAM optimization techniques for consumer GPUs during Flux training |
| DJZ AssassinKahb Flux Dev Test Lora Collection — Civitai model 654919 | Benchmark Flux LoRA model and reference for training results |