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/setup.py
+++ b/setup.py
@@ -86,7 +86,7 @@ setup(
            "fused-dense-lib  @ git+https://github.com/Dao-AILab/flash-attention@v2.6.2#subdirectory=csrc/fused_dense_lib",
        ],
        "deepspeed": [
-            "deepspeed @ git+https://github.com/microsoft/DeepSpeed.git@bc48371c5e1fb8fd70fc79285e66201dbb65679b",
+            "deepspeed==0.14.4",
            "deepspeed-kernels",
        ],
        "mamba-ssm": [