Avoids repeated work.#1163

Merged

zhuzilin merged 1 commit intoTHUDM:mainfrom

Dec 21, 2025

Commits on Dec 20, 2025

Avoids repeated work: mpu.get_context_parallel_rank() and -args.kl_coef are computed once outside the loop.
qqwqqw689
authored