From 949f210feba96368a7a98bc4eb2da5ede1348fe5 Mon Sep 17 00:00:00 2001 From: Helen Ngo Date: Mon, 2 Feb 2026 09:26:55 -0800 Subject: [PATCH] Fix seq pack flag in get_logprobs --- megatron/rl/rl_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/megatron/rl/rl_utils.py b/megatron/rl/rl_utils.py index 364a80db81e..352e3386bf2 100644 --- a/megatron/rl/rl_utils.py +++ b/megatron/rl/rl_utils.py @@ -999,7 +999,7 @@ def logprobs_forward_step(data_iterator, model, is_correction, packing_context=N b_trajs.cuda(), b_posids.cuda(), no_grad=True, - sequence_packing=b_packed_seq_params is not None, + sequence_packing=packing_context is not None, packed_seq_params=b_packed_seq_params, ), None,