need to update deepspeed version in extras too (#2161) [skip ci]

* need to update deepspeed version in extras too * fix patch import * fix monkeypatch reloading in tests and deepspeed patch * remove duplicated functionality fixture * reset LlamaForCausalLM too in fixtures for cce patch * reset llama attn too * disable xformers patch for cce * skip problematic test on low usage functionality
2024-12-09 14:01:44 -05:00
parent 5d6b088997
commit ab4b32187d
10 changed files with 60 additions and 45 deletions
--- a/setup.py
+++ b/setup.py
@@ -125,7 +125,7 @@ setup(
            "flash-attn==2.7.0.post2",
        ],
        "deepspeed": [
-            "deepspeed==0.15.4",
+            "deepspeed==0.16.1",
            "deepspeed-kernels",
        ],
        "mamba-ssm": [