Add accelerate dep

2023-05-30 16:36:13 -04:00
parent 13b80937f9
commit 36ec6e1a0e
1 changed files with 1 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
 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