multipack sampler support from openchat

2023-07-15 08:01:33 -04:00
parent 168a7a09cc
commit 81d60e96f0
3 changed files with 286 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,6 @@
 peft @ git+https://github.com/huggingface/peft.git
 transformers @ git+https://github.com/huggingface/transformers.git
 bitsandbytes>=0.39.0
-accelerate
 addict
 fire
 PyYAML==6.0
@@ -18,3 +17,4 @@ evaluate==0.4.0
 rouge-score==0.1.2
 scipy
 scikit-learn==1.2.2
+numba