FLUX.2 klein LoRA training with ai-toolkit: setup guide

Setting up ai-toolkit to train a FLUX.2 klein LoRA: support status, the settings known to work, and links to the official docs. Status in ai-toolkit: supported (klein base 4B and 9B).

About ai-toolkit

Ostris' trainer with a web UI and YAML configs; widest day-0 support for new image models. Community threads call it the easiest option for Flux/Krea. Sources: github.com/ostris/ai-toolkit, www.reddit.com/r/StableDiffusion/comments/1vrkd5x/which_platform_shoul

Repo	ostris/ai-toolkit
GitHub stars	12321
Last commit	2026-10-16
Latest release	none published

VRAM

12-16 GB

Settings that work for FLUX.2 klein (any trainer)

No verified settings yet.

General tips

Gated Hugging Face models: click 'Agree and access' on each model page, or you'll get 403 errors
Steps scale with dataset size (steps = epochs x images at batch 1)
Common safe start: LoKr with auto rank (~64), batch/grad accumulation 4 [Discord community reports (AI Toolkit / banodoco / lodestone_rock servers), Oct 2026]

Compare with other trainers

Where to rent a 16 GB+ GPU

Cheapest 16 GB+ cards (2026-10-16 14:43 UTC) from RunPod and Vast.ai. Prices move constantly; see the live price table.

GPU	VRAM	RunPod community $/hr	Vast.ai cheapest $/hr
RTX A4000	16 GB	$0.17	$0.09
Tesla V100	16 GB	$0.19	$0.10
RTX 3090	24 GB	$0.22	$0.12
RTX 5060 Ti	16 GB	—	$0.12
RTX 5070 Ti	16 GB	—	$0.12
RTX A5000	24 GB	$0.16	—

Sources

FAQ

Does ai-toolkit support FLUX.2 klein?

Yes: supported (klein base 4B and 9B) (see sources below).

How much VRAM do I need for a FLUX.2 klein LoRA?

12-16 GB

Weekly newsletter: new models, trainer releases, cheapest GPUs. Subscribe

Data updated 2026-10-16. Disclosure: RunPod and Vast.ai links are referral links; ImageLoRA may earn a commission at no cost to you. unknown = not yet verified from a source.