update README w deepspeed info (#605)

2023-09-22 00:15:52 -04:00
parent d5f8589021
commit c25ba7939b
6 changed files with 114 additions and 88 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,6 +7,7 @@ peft @ git+https://github.com/huggingface/peft.git
 transformers @ git+https://github.com/huggingface/transformers.git
 bitsandbytes>=0.41.1
 accelerate @ git+https://github.com/huggingface/accelerate
+deepspeed
 addict
 evaluate
 fire