AI Potluck
Model components / Fine-tuning code

AI Toolkit

Ostris

AI Toolkit is a training suite for fine-tuning diffusion models, covering image, instruction-edit, video and audio model families including FLUX.1, FLUX.2, Qwen-Image and Wan. It trains LoRA and LyCORIS adapters, ships a web UI for configuring and monitoring runs, and provides recipes for running on RunPod and Modal.

Verified 2026-08-31 via the repository LICENSE and README.

Openness

5 high confidence
5.0
license
MIT(OSI)
source
public(ostris/ai-toolkit)
core-gated
ungated(no enterprise or ee path in the repository root

MIT, an OSI licence, with the full source public and the shipped code in this repository. The LICENSE body was read in full: the stock MIT text, with no appended condition. No enterprise or ee path in the repository root; the readme's ostris cloud section offers rented gpus to fund development and an affiliate link, not a paid tier of the software.

Adoption

3 low confidence
3.0

No package registry carries this product, so stars are the only available signal and the band is capped at 3 by the instrument. 11,860 stargazers places it at level 3. Stars measure attention rather than use, which is why the confidence is low.

Capability

3 medium confidence
3.0

A diffusion-specialist trainer with wide and current model coverage but an adapter-centric method set and no distributed-training story. It is scored against onetrainer rather than the category anchor because megatron-lm's parallelism-and-scale ladder says nothing about a consumer-GPU diffusion trainer; 3 is where this category puts a solid method catalogue at limited scale, as it does for litgpt and torchtune.

  • https://raw.githubusercontent.com/ostris/ai-toolkit/main/README.md recorded 2026-08-31

    Supported Models section lists FLUX.1, FLUX.2 dev and klein-base 4B/9B, Flex.1, Flex.2, Chroma, Lumina2 and Qwen-Image under Image, with further Instruction/Edit, Video, Audio and Experimental groups; the README documents LoRA and LyCORIS training and an AI Toolkit UI.

Verified 2026-08-31