[misc] fix: Improve compare.py robustness for multi-GPU and vocab-padded models#2647

ko3n1g merged 1 commit intor0.3.0NVIDIA-NeMo/Megatron-Bridge:r0.3.0from

yuya/cherry-pick-compare-fix-r0.3.0NVIDIA-NeMo/Megatron-Bridge:yuya/cherry-pick-compare-fix-r0.3.0Copy head branch name to clipboard