DeepSpeed
DeepSpeed copied to clipboard
Remove deprecated `torch._six` imports
Closes #2845.
@microsoft-github-policy-service agree
@yasyf we currently recommend users use torch 1.8+, it appears this older version of torch does not allow from torch import inf. Perhaps we can change these imports to be dependent on torch version? Not sure what version of torch this was supported, I just tried on 1.13 and it works there.
@jeffra godo catch! it's only a problem if you're using pytorch 2.0.0+ (nightly). I'll make it conditional
@jeffra test failure seems unrelated?