update docker to compile latest bnb to properly support qlora

This commit is contained in:
Wing Lian
2023-05-27 12:36:53 -04:00
parent 35af017001
commit 312b8d51d6
3 changed files with 21 additions and 1 deletions

View File

@@ -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