Files
axolotl/requirements-tests.txt
Wing Lian 1ef70312ba fix optimizer reset for relora sft (#1414)
* fix optimizer reset

* set states to reset for 8bit optimizers and handle quantile runtime error for embeddings

* fix relora test to check grad_norm

* use flash attn for relora and tweak hyperparams for test

* fix messages field for test dataset
2024-12-03 08:58:23 -05:00

6 lines
54 B
Plaintext

pytest
pytest-xdist
pytest-retry
pytest-sugar
tbparse