ai-toolkit/todo_multigpu.md at e6180d1e1df9bfd520649ad08f03fcc3301b73f9 - ai-toolkit - Gitea: Git with a cup of tea

github/ai-toolkit

Files

Jaret Burkett 5e663746b8 Working multi gpu training. Still need a lot of tweaks and testing.

2025-01-25 16:46:20 -07:00

160 B

Raw Blame History

only do ema on main device? shouldne be needed other than saving and sampling
check when to unwrap model and what it does
disable timer for non main local