update docker to compile latest bnb to properly support qlora

2023-05-27 12:36:53 -04:00
parent 35af017001
commit 312b8d51d6
3 changed files with 21 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,3 +11,9 @@ sentencepiece
 wandb
 einops
 xformers
+# qlora things
+bert-score==0.3.13
+evaluate==0.4.0
+rouge-score==0.1.2
+scipy
+scikit-learn==1.2.2