add support for opimum bettertransformers

2023-05-27 17:57:29 -04:00
parent a6f5e5eaec
commit adea682316
5 changed files with 48 additions and 22 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