From 9e3f0cb5a79bd82ab9a17b0e6d30a43c399708de Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 10 Jan 2024 00:50:04 -0500 Subject: [PATCH] pin accelerate for deepspeed fix (#1080) --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index fed3d53f56..33ddd395df 100644 --- 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