[OMNIML-6082] [3/3] Write a sharded model's HF checkpoint shards from each tensor's owner rank - #2751

[OMNIML-6082] [3/3] Write a sharded model's HF checkpoint shards from each tensor's owner rank#2751

shengliangxu merged 6 commits into

mainNVIDIA/Model-Optimizer:mainfrom

shengliangx/hfep-12b-writer-piecesNVIDIA/Model-Optimizer:shengliangx/hfep-12b-writer-piecesCopy head branch name to clipboard