Z-Image LoRA training with ai-toolkit: setup guide

Setting up ai-toolkit to train a Z-Image LoRA: support status, the settings known to work, and links to the docs. Status in ai-toolkit: supported (Z-Image, Turbo, De-Turbo).

About ai-toolkit

Ostris' trainer with a web UI and YAML configs; widest day-0 support for new image models, and 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	12373
Last commit	2026-10-19
Latest release	none published

VRAM

Not published. musubi: 12GB+ recommended for image training (general); --fp8_llm saves VRAM

Settings that work for Z-Image (any trainer)

diffusers v0.42.0 (Oct 23, PR #14163) loads LoKr adapters for Z-Image

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)
A safe start: LoKr with auto rank (~64), batch/grad accumulation 4
Leave out of your captions whatever you want burned into the LoRA; caption what should stay changeable
ai-toolkit dataset folder weights let you train the same images with long, medium and short captions in separate folders
ai-toolkit supports validation loss, which shows when a run stops improving
In ai-toolkit, Cache text embeddings works for basically every model and frees the text encoder from VRAM and RAM
Prefer int8 convrot over qfloat8 quantization in ai-toolkit: better quality at the same memory
Gradient accumulation and batch size don't scale the learning rate in ai-toolkit; a bigger batch needs about 1.5-2.5x the steps

Compare with other trainers

Where to rent a 16 GB+ GPU

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

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

Sources

FAQ

Does ai-toolkit support Z-Image?

Yes: supported (Z-Image, Turbo, De-Turbo) (see sources below).

How much VRAM do I need for a Z-Image LoRA?

Not published. musubi: 12GB+ recommended for image training (general); --fp8_llm saves VRAM

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

Data updated 2026-10-20. Disclosure: RunPod and Vast.ai links are referral links; ImageLoRA may earn a commission at no cost to you. unknown = no data yet. · Raw data (JSON) · llms.txt