DeepSpeed

c8c57b8c - add sharded loading for safetensors in AutoTP (#4854)

add sharded loading for safetensors in AutoTP (#4854) Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>