Merge pull request #114 from OpenAccess-AI-Collective/accelerate-dep

Add accelerate dep
2023-05-31 00:47:17 -04:00
parent 0e4be625ae 36ec6e1a0e
commit dcb03d6da4
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