[ckpt] fix: Use DTensor split shapes for Megatron-FSDP TP loading#3746
Open
conver334 wants to merge 1 commit into
Open
[ckpt] fix: Use DTensor split shapes for Megatron-FSDP TP loading#3746conver334 wants to merge 1 commit into
conver334 wants to merge 1 commit into
