Skip to content

Pull requests: ostris/ai-toolkit

New pull request New
34 Open 191 Closed
34 Open 191 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 ACE Step Training: float32 saving and shape fix
#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…
1 of 11 tasks
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 orakulstorm-hue 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…
1
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…
Training, Optimizers, Device Handling & Z-Image Improvements
#713 opened Feb 13, 2026 by daanforever • Draft
Add logging support for Comet
#709 opened Feb 11, 2026 by sebastianberns Loading…
1
Feature/prodigy optimizers and adamw for Z Image Lora
#706 opened Feb 9, 2026 by whmc76 Loading…
install loras directly to file path
#705 opened Feb 7, 2026 by riffmaster-2001 Loading…
Implement polynomial learning rate scheduler
#702 opened Feb 6, 2026 by NBSTpeterhill Loading…
AdamWFP8 support
#699 opened Feb 5, 2026 by saluzafa • Draft
1
feat: Add Continue Training feature for completed jobs
#698 opened Feb 4, 2026 by diodiogod Loading…
ProTip! Adding no:label will show everything without a label.