This website requires JavaScript.
Explore
Help
Register
Sign In
ostris
/
ai-toolkit
Watch
1
Star
0
Fork
0
You've already forked ai-toolkit
mirror of
https://github.com/ostris/ai-toolkit.git
synced
2026-02-25 06:43:57 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
14ccf2f3cea98aa70436f0553f0a5788ca55e7f8
ai-toolkit
/
extensions_built_in
/
diffusion_models
/
hidream
/
src
/
models
History
Jaret Burkett
524bd2edfc
Make flash attn optional. Handle larger batch sizes.
2025-04-14 14:34:46 +00:00
..
transformers
Hidream is training, but has a memory leak
2025-04-13 23:28:18 +00:00
attention_processor.py
Make flash attn optional. Handle larger batch sizes.
2025-04-14 14:34:46 +00:00
attention.py
Initial support for hidream. Still a WIP
2025-04-13 13:50:11 -06:00
embeddings.py
Initial support for hidream. Still a WIP
2025-04-13 13:50:11 -06:00
moe.py
Remove some moe stuff for finetuning. Drastically reduces vram usage
2025-04-14 00:57:34 +00:00
