ostris / ai-toolkit
git clone https://github.com/ostris/ai-toolkit.git
The ultimate training toolkit for finetuning diffusion models
+5
stars, 24h · #6,332
+~70
stars, 7 days · #2,990
+~300
stars, 30 days · #2,713
12k
★ total · +1% this week
1.6k
forks · 90 open issues
Star history (as recorded here)
LanguagePython
LicenseMIT
Created2023-07-05 (3y ago)
Last pushtoday
Watchers105
OwnerOrganization
Links
README excerpt
# Ostris AI Toolkit AI Toolkit is an easy to use all in one training suite for diffusion models. I try to support all the latest models on consumer grade hardware. Image and video models. It can be run as a GUI or CLI. It is designed to be easy to use but still have every feature imaginable. Free and open source. ## Supported Models ### Image - black-forest-labs/FLUX.1-dev (FLUX.1) - black-forest-labs/FLUX.2-dev (FLUX.2) - black-forest-labs/FLUX.2-klein-base-4B (FLUX.2-klein-base-4B) - black-forest-labs/FLUX.2-klein-base-9B (FLUX.2-klein-base-9B) - black-forest-labs/flux-3-image-klein-base (FLUX.3 Image Klein) - ostris/Flex.1-alpha (Flex.1) - ostris/Flex.2-preview (Flex.2) - lodestones/Chroma1-Base (Chroma) - Alpha-VLLM/Lumina-Image-2.0 (Lumina2) - Qwen/Qwen-Image (Qwen-Image) - Qwen/Qwen-Image-2512 (Qwen-Image-2512) - Qwen/Qwen-Image-2.1 (Qwen-Image-2.1) - Qwen/Qwen-Image-2.1-Turbo (Qwen-Image-2.1 Turbo) - HiDream-ai/HiDream-I1-Full (HiDream I1) - OmniGen2/OmniGen2 (OmniGen2) - Tongyi-MAI/Z-Image-Turbo (Z-Image Turbo) - Tongyi-MAI/Z-Image (Z-Image) - ostris/Z-Image-De-Turbo (Z-Image De-Turbo) - zhen-nan/L2P (Z-Image L2P) - stabilityai/stable-diffusion-xl-base-1.0 (SDXL) - stable-diffusion-v1-5/stable-diffusion-v1-5 (SD 1.5) - baidu/ERNIE-Image (ERNIE-Image) - NucleusAI/Nucleus-Image (Nucleus-Image) - Boogu/Boogu-Image-0.1-Base (Boogu Image 0.1) - HiDream-ai/HiDream-O1-Image (HiDream O1) - ideogram-ai/ideogram-4-fp8 (Ideogram 4 FP8) - Photoroom/prxpixel-t2i (PRXPixel) - speridlabs/iris-3b (Iris-3B) - circlestone-labs/Anima-Base-v1.0-Diffusers (Anima) - krea/Krea-2-Raw (Krea 2) - krea/Krea-2-Turbo (Krea 2 Turbo) - inclusionAI/Ming-Image-0.1-Design (Ming-Image 0.1 Design) - microsoft/Mage-Flow-Base (Mage-Flow) - OpenMOSS-Team/MOSS-Image-SFT (MOSS-Image SFT) - OpenMOSS-Team/MOSS-Image-Base (MOSS-Image Base) ### Instruction / Edit - black-forest-labs/FLUX.1-Kontext-dev (FLUX.1-Kontext-dev) - Qwen/Qwen-Image-Edit (Qwen-Image-Edit) - Qwen/Qwen-Image-Edit-2509 (Qwen-Image-Edit-2509) - Qwen/Qwen-Image-Edit-2511 (Qwen-Image-Edit-2511) - Qwen/Qwen-Image-2.1 (Qwen-Image-2.1) - one model for both; it edits when your dataset has control images - Qwen/Qwen-Image-2.1-Turbo (Qwen-Image-2.1 Turbo) - same, with the 8-step turbo weights - HiDream-ai/HiDream-E1-1 (HiDream E1) - Boogu/Boogu-Image-0.1-Edit (Boogu Image Edit) - krea/Krea-2-Raw (Krea 2 Edit Training) - krea/Krea-2-Turbo (Krea 2 Turbo Edit Training) - microsoft/Mage-Flow-Edit-Base (Mage-Flow Edit) ### Video - Wan-AI/Wan2.1-T2V-1.3B-Diffusers (Wan 2.1 1.3B) - Wan-AI/Wan2.1-I2V-14B-480P-Diffusers (Wan 2.1 I2V 14B-480P) - Wan-AI/Wan2.1-I2V-14B-720P-Diffusers (Wan 2.1 I2V 14B-720P) - Wan-AI/Wan2.1-T2V-14B-Diffusers (Wan 2.1 14B) - Wan-AI/Wan2.2-T2V-A14B-Diffusers (Wan 2.2 14B) - Wan-AI/Wan2.2-I2V-A14B-Diffusers (Wan 2.2 I2V 14B) - Wan-AI/Wan2.2-TI2V-5B-Diffusers (Wan 2.2 TI2V 5B) - Lightricks/LTX-2 (LTX-2) - Lightricks/LTX-2.3 (LTX-2.3) - Lightricks/LTX-2.5 (LTX-2.5) - MiniMaxAI/MiniMax-H3 (MiniMaxAI/MiniMax