[core] Shard tensor-parallel checkpoints on load and save - #14544

[core] Shard tensor-parallel checkpoints on load and save#14544

JingyaHuang wants to merge 17 commits into

huggingface:mainhuggingface/diffusers:mainfrom

JingyaHuang:add-shard-ckpt-loadingJingyaHuang/diffusers:add-shard-ckpt-loadingCopy head branch name to clipboard