README

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)
- briaai/Fibo-2.0-dev (Fibo 2.0 Dev)

 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)
- PrunaAI/P-H3-Image (P-H3-Image) - text-to-image and editing with up to 8 reference images; it edits when your dataset has control images

 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/Wan