ostris/ai-toolkit

View on GitHub

The ultimate training toolkit for finetuning diffusion models

Is ostris/ai-toolkit beginner-friendly?

In the last 30 days, ostris/ai-toolkit merged 2 of 16 pull requests from outside contributors, and its maintainers usually reply within 1 day. 8 beginner-friendly issues are open now.

Stars: 12.2k
Forks: 1.6k
Open beginner issues: 8
Indexed issues: 69
Avg merge: 14h 42m
Merged PRs (30d): 2

Dominant language: Python
License: MIT
Last GitHub push: Sep 27, 2026
Latest indexed: Oct 4, 2026
Contributing guide: No contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

How to contribute to ostris/ai-toolkit

There is no contributing guide, so start from the README and a few recently merged pull requests to see how changes are proposed.
Your contributions will be published under the project's MIT license.
Pick one of the 8 open beginner-friendly issues below and comment that you want to work on it before you start.

Beginner friendly Everything

Open All

Issues somebody may already be working on are listed last. List everything by date

Bug: Prodigy 8-bit optimizer state fails to restore due to Auto8bitTensor deserialization error
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 72/100

ostris/ai-toolkit#1076 · Opened Oct 10, 2026

Maintainers usually reply within 1 day
Qwen-Image-2.1 LoRA: img_mlp lora_B stays exactly zero in all checkpoints (attention adapters train fine)
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 25/100

ostris/ai-toolkit#1075 · 1 comment · Opened Oct 10, 2026

Maintainers usually reply within 1 day
convrot int8 problems (windows)
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 35/100

ostris/ai-toolkit#1074 · Opened Oct 7, 2026

Maintainers usually reply within 1 day
Minimax-H3, create LoRA for Text Encoder - for a very long time
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 25/100

ostris/ai-toolkit#1071 · 2 comments · Opened Oct 2, 2026

Maintainers usually reply within 1 day
Krea 2 train on RAW preview on Turbo
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 45/100

ostris/ai-toolkit#1070 · Opened Oct 1, 2026

Maintainers usually reply within 1 day
Chroma training slowdown after update: ~5 s/it → 130–300 s/it until job is recreated
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 25/100

ostris/ai-toolkit#1066 · Opened Sep 26, 2026

Maintainers usually reply within 1 day
YuE2: SheetSage2 remote code is loaded unpinned, and the ABC repair path skips upstream's chord spelling
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 70/100

ostris/ai-toolkit#1064 · Opened Sep 25, 2026

Maintainers usually reply within 1 day
Microsoft ADPCM WAVs load at a fraction of their length (audio comes out sped up) through torchaudio.load
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 38/100

ostris/ai-toolkit#1063 · Opened Sep 24, 2026

Maintainers usually reply within 1 day
Stop Job CAN resume later. not working
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 45/100

ostris/ai-toolkit#1058 · 3 comments · 1 reaction · Opened Sep 22, 2026

Maintainers usually reply within 1 day
Krea 2 - Qwen2.1 LoRA + Layer Offloading bug
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 22/100

ostris/ai-toolkit#1057 · 1 comment · Opened Sep 21, 2026

Maintainers usually reply within 1 day
Bug: Zero-gradient deadlock when expanding LoRA/DoRA rank (New capacity remains completely unused)
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 78/100

ostris/ai-toolkit#1056 · 1 reaction · Opened Sep 21, 2026

Maintainers usually reply within 1 day
Dataset caches thumbnail previews even when replacing them
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 55/100

ostris/ai-toolkit#1051 · Opened Sep 18, 2026

Maintainers usually reply within 1 day
MiniMax-H3 ref2va training: clean hang on first step with triton; OOM at load without triton (RTX 5090 32GB, Windows, torch 2.14+cu130)
Open
- Is something wrong?
Difficulty 5/5 Over a week Newbie friendliness 15/100

ostris/ai-toolkit#1050 · Opened Sep 18, 2026

Maintainers usually reply within 1 day
`split_model_over_gpus` + LoRA training on multi-GPU: five independent device/signature bugs break FLUX LoRA runs (16 GB cards)
Open
- Is something wrong?
Difficulty 5/5 Over a week Newbie friendliness 15/100

ostris/ai-toolkit#1048 · Opened Sep 17, 2026

Maintainers usually reply within 1 day
One stuck `running` job row permanently clogs the queue — worker trusts exit code 0 and never reaps dead/hung jobs
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 45/100

ostris/ai-toolkit#1047 · Opened Sep 17, 2026

Maintainers usually reply within 1 day
Auto-captioning audio fails with 'NoneType' object has no attribute 'device' in low-VRAM mode
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 45/100

ostris/ai-toolkit#1043 · Opened Sep 15, 2026

Maintainers usually reply within 1 day
Krea 2: q/k/v dtypes can diverge under gradient checkpointing, breaking backward
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 65/100

ostris/ai-toolkit#1041 · Opened Sep 13, 2026

Maintainers usually reply within 1 day
ever since the update z-image lora training stopped working
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 18/100

ostris/ai-toolkit#1038 · Opened Sep 11, 2026

Maintainers usually reply within 1 day
Windows: stale cron worker survives UI shutdown, breaks next startup with EPERM on prisma generate
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 78/100

ostris/ai-toolkit#1033 · Opened Sep 9, 2026

Maintainers usually reply within 1 day
ai-toolkit 0.3.6 starts ok - New job start - Lora creation - Krea-2 (raw) - but it is VERY slow. Sometimes it works. No changes.
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 15/100

ostris/ai-toolkit#1032 · 3 comments · Opened Sep 8, 2026

Maintainers usually reply within 1 day
Bug Report: Ideogram 4 training fails during transformer dequantization
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 25/100

ostris/ai-toolkit#1030 · 2 comments · Opened Sep 5, 2026

Maintainers usually reply within 1 day
train_lora_omnigen2_24gb.yaml: unterminated string breaks YAML parsing
Open
- Is something wrong?
Difficulty 1/5 Under an hour Newbie friendliness 90/100

ostris/ai-toolkit#1029 · Opened Sep 4, 2026

Maintainers usually reply within 1 day
[Bug] training_adapter field ignored in headless yaml for Z‑Image‑Turbo
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 45/100

ostris/ai-toolkit#1027 · Opened Sep 3, 2026

Maintainers usually reply within 1 day
LTXV Audio Training even though do_audio is set to False
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 50/100

ostris/ai-toolkit#1020 · 4 comments · Opened Aug 27, 2026

Maintainers usually reply within 1 day
Anima res 1280 - Sizes of tensors must match except in dimension 3
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 32/100

ostris/ai-toolkit#1019 · Opened Aug 25, 2026

Maintainers usually reply within 1 day
Crash training video when offload active
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 25/100

ostris/ai-toolkit#1013 · Opened Aug 14, 2026

Maintainers usually reply within 1 day
torch.compile fails with record_stream after commit d184c6c
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 35/100

ostris/ai-toolkit#1012 · Opened Aug 13, 2026

Maintainers usually reply within 1 day
New Version is 14x slower then the one with the torchao error
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 15/100

ostris/ai-toolkit#1007 · 8 comments · Opened Aug 11, 2026

Maintainers usually reply within 1 day
Just very tiny thing: "Cache Text Embeddings" Info balloon needs an update
Open
- Is something wrong?
Difficulty 1/5 Under an hour Newbie friendliness 65/100

ostris/ai-toolkit#1006 · 1 comment · Opened Aug 11, 2026

Maintainers usually reply within 1 day
80GB of virtual memory needed to train Klein 9B
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 25/100

ostris/ai-toolkit#1003 · 4 comments · Opened Aug 9, 2026

Maintainers usually reply within 1 day
MiniMax-H3: official non-pruned transformer weights fail to load on 0.12.8
Open
- Is something wrong?
Difficulty 3/5 Half a day Newbie friendliness 55/100

ostris/ai-toolkit#1001 · 1 comment · Opened Aug 8, 2026

Maintainers usually reply within 1 day
get_lr_scheduler's diffusers fallback cannot succeed: both reachable SchedulerTypes fail on the injected total_iters
Open
- Is something wrong?
Difficulty 3/5 Half a day Newbie friendliness 58/100

ostris/ai-toolkit#996 · Opened Aug 7, 2026

Maintainers usually reply within 1 day
torchcodec version 0.9.1 is no longer compatible with PyTorch version 2.13.0. I hope the author fixes this and also checks for potential issues with environment dependencies for other specified versions.
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 40/100

ostris/ai-toolkit#994 · 1 comment · Opened Aug 7, 2026

Maintainers usually reply within 1 day
Qwen image edit 2509 sigkill
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 15/100

ostris/ai-toolkit#993 · 1 comment · Opened Aug 5, 2026

Maintainers usually reply within 1 day
Out of Memory at "Generating baseline samples before training"
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 15/100

ostris/ai-toolkit#992 · Opened Aug 4, 2026

Maintainers usually reply within 1 day
H3 loRA fine-tuning but generated videos are pure noise
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 20/100

ostris/ai-toolkit#991 · 10 comments · 4 reactions · Opened Aug 4, 2026

Maintainers usually reply within 1 day
Severe VRAM regression / OOM with Krea 2 training after PyTorch 2.13 update
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 20/100

ostris/ai-toolkit#990 · 7 comments · 1 reaction · Opened Aug 3, 2026

Maintainers usually reply within 1 day
FLUX.2 Klein 9B qfloat8 block quantization repeatedly triggers CUDA unknown/GPU loss during CPU→GPU block transfers on RTX 5090 + WSL2
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 20/100

ostris/ai-toolkit#989 · Opened Aug 1, 2026

Maintainers usually reply within 1 day
[SOLVED] LECO Slider LoRA Training Is Impossible Due to Two Year Old 'NoneType' object has no attribute Bug
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 22/100

ostris/ai-toolkit#983 · 1 reaction · Opened Jul 28, 2026

Maintainers usually reply within 1 day
Multiple checkpoints deleted automatically with continued training
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 65/100

ostris/ai-toolkit#981 · 3 comments · Opened Jul 28, 2026

Maintainers usually reply within 1 day
About the pixel alignment task of training krea2 edit lora
Open
- Is something wrong?
Difficulty 5/5 Over a week Newbie friendliness 15/100

ostris/ai-toolkit#980 · Opened Jul 28, 2026

Maintainers usually reply within 1 day
Mage-Flow LoRA training fails with batch_size > 1 (LoRA multiplier assumes batch is dim 0)
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 70/100

ostris/ai-toolkit#977 · Opened Jul 26, 2026

Maintainers usually reply within 1 day
Stuck "Stopping job..." after power loss / dead PID — stop handler never marks job stopped
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 65/100

ostris/ai-toolkit#976 · 1 comment · Opened Jul 25, 2026

Maintainers usually reply within 1 day
LTX2 LoRA training completes successfully but generated LoRA contains only zero weights
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 20/100

ostris/ai-toolkit#975 · Opened Jul 25, 2026

Maintainers usually reply within 1 day
AttributeError: 'NoneType' object has no attribute 'dataset_config'
Open
- Is something wrong?
Difficulty 3/5 1-2 days Newbie friendliness 52/100

ostris/ai-toolkit#973 · Opened Jul 24, 2026

Maintainers usually reply within 1 day
convrot8 qtype_te fails with 3D tensor during text embedding caching on Windows (Qwen3-VL text encoder)
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 64/100

ostris/ai-toolkit#971 · Opened Jul 23, 2026

Maintainers usually reply within 1 day
Why does the UI store config in a database instead of a yaml file in the config directory
Open
- Is something wrong?
Difficulty 5/5 Over a week Newbie friendliness 18/100

ostris/ai-toolkit#970 · 4 comments · Opened Jul 23, 2026

Maintainers usually reply within 1 day
alpha type mismatch when alpha is a scalar (in result)
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 55/100

ostris/ai-toolkit#969 · Opened Jul 22, 2026

Maintainers usually reply within 1 day
Attempting to train Krea2 causes fatal error "Job process terminated with exit code 1."
Open
- Is something wrong?
Difficulty 2/5 1-3 hours Newbie friendliness 45/100

ostris/ai-toolkit#968 · 1 comment · Opened Jul 21, 2026

Maintainers usually reply within 1 day
CUDA error: out of memory
Open
- Is something wrong?
Difficulty 4/5 3-5 days Newbie friendliness 22/100

ostris/ai-toolkit#929 · 2 comments · 1 reaction · Opened Jul 3, 2026

Maintainers usually reply within 1 day

ostris/ai-toolkit

Is ostris/ai-toolkit beginner-friendly?

How to contribute to ostris/ai-toolkit

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Is something wrong?

Get new issues in your inbox