Bump deepspeed 20240727 (#1790)

* pin deepspeed to 0.14.4 otherwise it doesn't play nice with trl * Add test to import to try to trigger import dependencies
2024-07-27 10:24:11 -04:00
parent 6a9cfec222
commit 22680913f3
3 changed files with 22 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ transformers==4.43.1
 tokenizers==0.19.1
 bitsandbytes==0.43.1
 accelerate==0.32.0
-deepspeed @ git+https://github.com/microsoft/DeepSpeed.git@bc48371c5e1fb8fd70fc79285e66201dbb65679b
+deepspeed==0.14.4
 pydantic==2.6.3
 addict
 fire