DeepSpeed

[zero3] release tmp memory when consolidating fp16 weights

#1220

Merged