Add: mlflow for experiment tracking (#1059) [skip ci]

* Update requirements.txt adding mlflow * Update __init__.py Imports for mlflow * Update README.md * Create mlflow_.py (#1) * Update README.md * fix precommits * Update README.md Update mlflow_tracking_uri * Update trainer_builder.py update trainer building * chore: lint * make ternary a bit more readable --------- Co-authored-by: Wing Lian <wing.lian@gmail.com>
2024-01-09 15:34:09 +01:00
parent 651b7a31fc
commit 090c24dcb0
5 changed files with 34 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -22,6 +22,7 @@ hf_transfer
 colorama
 numba
 numpy>=1.24.4
+mlflow
 # qlora things
 bert-score==0.3.13
 evaluate==0.4.0