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 4040_DEEPSPEED_AVAILABLE = (
4141 # DeepSpeed fails under 0.8.2 with torch 2.0: https://github.com/microsoft/DeepSpeed/pull/2863
4242 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+ )
4850)
4951if TYPE_CHECKING and _DEEPSPEED_AVAILABLE :
5052 import deepspeed
You can’t perform that action at this time.
0 commit comments