Consistency regularizer adapter using NxPenalties constraints.
Expects paired outputs in loss_fn_inputs (e.g., "original_logprobs").
Consistency regularizer adapter using NxPenalties constraints.
Expects paired outputs in loss_fn_inputs (e.g., "original_logprobs").