remove duplicated code in ZeRO (#2655)

Commit

2 years ago

remove duplicated code in ZeRO (#2655) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>