qwerrwe / src /axolotl /utils /callbacks.py

Commit History

improve GPU logging to break out pytorch cache and system mem
7b55fe6

tmm1 commited on

log GPU memory usage
e303d64

tmm1 commited on

more gpt-neox long ctx fixes
ab5cd28

winglian commited on

fix bettertransformers save, force it to skip after saving correctly in callback
1a82082

winglian commited on

more tweaks to do pre-training with bettertransformers
1210dc8

winglian commited on

Apply isort then black
37293dc

Nanobit commited on

Lint callbacks.py
5062eca

Nanobit commited on

black formatting
2bc1a5b

winglian commited on

Add callback save peft_model on_save
0d6708b

Nanobit commited on