Fix deepspeed tests (#3003)

* Unpin deepspeed * Include proper branch for docker image * Properly working * Revert all other changes
huggingface · Aug 9, 2024 · dc3b5ad · dc3b5ad
1 parent 12a5bef
commit dc3b5ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -36,7 +36,7 @@
     "timm",
 ]
 extras["testing"] = extras["test_prod"] + extras["test_dev"]
-extras["deepspeed"] = ["deepspeed<=0.14.0"]
+extras["deepspeed"] = ["deepspeed"]
 extras["rich"] = ["rich"]
 
 extras["test_trackers"] = ["wandb", "comet-ml", "tensorboard", "dvclive"]