remove unused outputs back to default

linkedin · Jan 22, 2025 · a18766c · a18766c
1 parent 1582cda
commit a18766c
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/src/liger_kernel/chunked_loss/fused_linear_preference.py b/src/liger_kernel/chunked_loss/fused_linear_preference.py
@@ -430,7 +430,5 @@ def _compute_loss(
             chosen_logits_mean,
             rejected_logits_mean,
             chosen_nll_loss,
-            ref_chosen_logps if use_ref_model else None,
-            ref_rejected_logps if use_ref_model else None,
         )
         return loss, (*return_vars, *aux_outputs)
diff --git a/test/utils.py b/test/utils.py
@@ -546,8 +546,6 @@ def get_batch_loss_metrics(
             policy_chosen_logits.detach().mean(),
             policy_rejected_logits.detach().mean(),
             policy_nll_loss,
-            ref_chosen_logps if self.use_ref_model else None,
-            ref_rejected_logps if self.use_ref_model else None,
         )
         return loss, (*return_vars, *aggregated_aux_outputs)