Ref : https://github.com/NVIDIA-NeMo/Megatron-Bridge/pull/643 Functional test in the PR old revisions. Same config runs fine locally but fails when run with the functional test infra.