Zach Nussbaum
|
6c4f449b7a
|
fix: update train scripts and configs for other models (#1164)
* feat: falcon config
* feat: mpt config
* chore: gitignore
* refactor: step calculation
* fix: attention mask + shuffle on epoch end
* fix: return tensors
* fix: wait for everyone
* chore: config
* chore: ds config
* fix: remove ccols
* fix: logging and saving
* chore: add einops
|
2023-07-12 15:18:24 -04:00 |
|