File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed
src/lightning/fabric/strategies Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change 40
40
_DEEPSPEED_AVAILABLE = (
41
41
# DeepSpeed fails under 0.8.2 with torch 2.0: https://github.com/microsoft/DeepSpeed/pull/2863
42
42
RequirementCache ("deepspeed>=0.8.2" )
43
- or not _TORCH_GREATER_EQUAL_2_0
44
- # check packaging because of https://github.com/microsoft/DeepSpeed/pull/2771
45
- # remove the packaging check when min version is >=0.8.1
46
- and RequirementCache ("deepspeed" )
47
- and RequirementCache ("packaging>=20.0" )
43
+ or (
44
+ not _TORCH_GREATER_EQUAL_2_0
45
+ and RequirementCache ("deepspeed" )
46
+ # check packaging because of https://github.com/microsoft/DeepSpeed/pull/2771
47
+ # remove the packaging check when min version is >=0.8.1
48
+ and RequirementCache ("packaging>=20.0" )
49
+ )
48
50
)
49
51
if TYPE_CHECKING and _DEEPSPEED_AVAILABLE :
50
52
import deepspeed
You can’t perform that action at this time.
0 commit comments