Update requirements/pytorch/strategies.txt

Lightning-AI · Jul 26, 2022 · 1318cdc · 1318cdc
1 parent cd9f135
commit 1318cdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/pytorch/strategies.txt b/requirements/pytorch/strategies.txt
@@ -1,5 +1,5 @@
 fairscale>=0.4.5, <=0.4.6
-deepspeed<0.6.5
+deepspeed>=0.6.0, <0.6.5
 # no need to install with [pytorch] as pytorch is already installed
 horovod>=0.21.2, !=0.24.0, <0.25.1
 hivemind>=1.0.1, <=1.0.1; sys_platform == 'linux'