Skip to content

Pull requests: ostris/ai-toolkit

New pull request New
32 Open 203 Closed
32 Open 203 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

fix(flux2-klein): support community Qwen3 text encoders with missing chat_template
#838 opened May 20, 2026 by smhc Loading…
fix(flux2-klein-base): clamp transformer residual stream to bf16-safe range
#837 opened May 20, 2026 by fengcunhan Loading…
1
Update dgx_instructions.md
#830 opened May 12, 2026 by NeoTech Loading…
Add dual-GPU model-parallel paths for 8 LoRA-training architectures (FLUX.2 / Wan 2.2 / LTX-2 / Qwen-Image / Chroma / FLUX.1-Kontext / HiDream-I1 / Z-Image)
#829 opened May 11, 2026 by genno-whittlery Loading…
1
6
Make UI fully mobile responsive
#828 opened May 10, 2026 by Gkrumbach07 Loading…
1
Fix Ace Step 1.5 Batching error
#825 opened May 9, 2026 by AdamNizol Loading…
Enable VAE tiling for Wan to avoid OOM at sample-time
#819 opened May 7, 2026 by jeremy-linchao Loading…
fix: ACE Step Training float32 saving and shape
#814 opened May 2, 2026 by DefinitlyEvil Loading…
perf: enable pin_memory on CUDA DataLoaders for faster CPU->GPU transfer
#813 opened May 1, 2026 by Booyaka101 Loading…
1
Add Florence-2 captioner to Auto Caption
#812 opened Apr 30, 2026 by Zorgsd Loading…
fix(ui): generate CSS for grid-cols-13 through grid-cols-40
#811 opened Apr 29, 2026 by kyleinprogress Loading…
RLHF Flow-GRPO implementation POC
#808 opened Apr 28, 2026 by ifilipis Loading…
8 of 10 tasks
2
fix: handle missing Flux VAE config values and None prompt inputs
#807 opened Apr 28, 2026 by octaviomejiadiaz Loading…
Add Rose optimizer
#805 opened Apr 26, 2026 by MatthewK78 Loading…
feat(ui): add job export/import functionality
#803 opened Apr 25, 2026 by SGr33n Loading…
Update manager_modules.py
#778 opened Apr 4, 2026 by OrakulStudio Loading…
Fix flux_train_ui startup on Gradio 6 by removing removed Image kwargs
#777 opened Apr 3, 2026 by iqdoctor Loading…
1
Add FSDP v2 multi-GPU parameter sharding for LoRA training
#774 opened Apr 2, 2026 by luke9705 Loading…
Push intermediate checkpoints to HuggingFace Hub
#766 opened Mar 29, 2026 by Rasaboun Contributor Loading…
14
Add MLflow experiment tracking
#765 opened Mar 29, 2026 by Rasaboun Contributor • Draft
10 tasks done
29
Add ComfyUI Workflows
#743 opened Mar 21, 2026 by Frankoloko Loading…
Adding support for Bria's FIBO model
#732 opened Mar 4, 2026 by kfirgoldberg Loading…
3
feat: enable immediate saving on UI stop and enhance optimizer state backup
#727 opened Feb 27, 2026 by avan06 Loading…
2
Seed expanded LoRA dimensions with noise instead of zeros
#724 opened Feb 23, 2026 by Stefatorus Loading…
4 of 5 tasks
Refactor file opening to use _open_text for encoding
#718 opened Feb 18, 2026 by marduk191 Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.