DeepSpeed

[zero3] release tmp memory when consolidating fp16 weights take2

#1223

Merged