[WIP] fix transforrmers api change at 5.2.0 #1432

Job	Run time
e2e-test-megatron (0, 8, test_qwen3_30B_A3B_r3.py)	19m 27s
e2e-test-megatron (8, test_qwen3_4B_ppo_train_critic_only.py)	25m 25s
e2e-test-megatron (8, test_qwen3_30B_A3B.py, 1, 1)	24m 45s
e2e-test-changed-detect	-1s
e2e-test-megatron (8, test_qwen3_4B_ppo.py)	13m 4s
e2e-test-megatron (8, test_quick_start_glm4_9B.py)	11m 36s
e2e-test-megatron (8, test_qwen2.5_0.5B_opd_sglang.py)	15m 27s
e2e-test-megatron (0, 8, test_qwen3_30B_A3B_r3.py, 1, 1)	21m 57s
e2e-test-megatron (8, test_moonlight_16B_A3B.py)	6m 45s
e2e-test-megatron (8, test_mimo_7B_mtp_only_grad.py)	4m 26s
e2e-test-short	-1s
e2e-test-image	-1s
e2e-test-megatron (0, 8, test_moonlight_16B_A3B_r3.py)	9m 13s
e2e-test-megatron (8, test_qwen2.5_0.5B_debug_rollout_then_train.py)	4m 53s
e2e-test-precision	-1s
e2e-test-ckpt	-1s
e2e-test-fsdp	-1s
e2e-test-changed	0s
	2h 36m 52s

Provide feedback