pin accelerate for deepspeed fix (#1080)

2024-01-10 00:50:04 -05:00
parent 2f2582e6ed
commit 9e3f0cb5a7
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ peft==0.7.0
 transformers @ git+https://github.com/huggingface/transformers.git@3cefac1d974db5e2825a0cb2b842883a628be7a0
 tokenizers==0.15.0
 bitsandbytes>=0.41.1
-accelerate==0.24.1
+accelerate @ git+https://github.com/huggingface/accelerate.git@0d2280dadc6a93413a5496613b7fdda3a4d2551b
 deepspeed
 addict
 fire