add support for opimum bettertransformers

2023-05-27 17:57:29 -04:00
parent 41e4f6ca31
commit 1edc30c786
5 changed files with 44 additions and 18 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,6 +11,7 @@ sentencepiece
 wandb
 einops
 xformers
+optimum
 # qlora things
 bert-score==0.3.13
 evaluate==0.4.0