ZImageModel Fix - ai-toolkit Extension (LoRA Keys)
Details
Download Files
Model description
ZImageModel extension for Ostris ai-toolkit. Fixes: - LoRA key conversion (Ostris → ZImage transformer layers) - attn_qkv/out, mlp_w1/w2 targets - Assistant LoRA merge support Install: Replace ai-toolkit/extensions_built_in/diffusion_models/z_image/z_image.py Compatible: Z-Image-Turbo, rank 8-32 LoRA.

