From 53845d25961b0a7dc98dd5c9b69430629941b237 Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Mon, 22 Jan 2024 11:47:01 +0100 Subject: [PATCH] Fix deepspeed issue (#2366) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index a7cc2a74b5f..ef8263c671d 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ extras["docs"] = [] extras["test_prod"] = ["pytest", "pytest-xdist", "pytest-subtests", "parameterized"] extras["test_dev"] = [ - "datasets", "evaluate", "transformers", "scipy", "scikit-learn", "deepspeed", "tqdm", "bitsandbytes", "timm" + "datasets", "evaluate", "transformers", "scipy", "scikit-learn", "deepspeed<0.13.0", "tqdm", "bitsandbytes", "timm" ] extras["testing"] = extras["test_prod"] + extras["test_dev"] extras["rich"] = ["rich"]