Fixes based on Nemotron3 tests #4750

Job	Run time
pr-gate / check-file-changes	21s
pr-gate / wait-checks / wait	12m 6s
gpu-tests (gpu_trtllm, 30, tensorrt-llm/release:1.3.0rc12)	11m 38s
gpu-tests (gpu_megatron, 45, nemo:26.04)	29m 5s
gpu-tests (gpu, 60, pytorch:26.03-py3)	55m 0s
gpu-pr-required-check	3s
	1h 48m 13s

Provide feedback