pin accelerate so it works with llama2 (#330)

2023-07-30 22:20:06 -04:00
parent 894cba09f3
commit 6c9a87c8ee
2 changed files with 1 additions and 5 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 peft @ git+https://github.com/huggingface/peft.git
 transformers @ git+https://github.com/huggingface/transformers.git
 bitsandbytes>=0.39.0
-accelerate==0.21.0
+accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b
 addict
 fire
 PyYAML==6.0