Fix for pytorch 2.0 compatibility in deepspeed#1

cbcase merged 1 commit intoaugmentcode:augmentaugmentcode/DeeperSpeed:augmentfrom

cbcase:pytorch2-testingcbcase/DeeperSpeed:pytorch2-testingCopy head branch name to clipboard