Update rl_replacements.py

2025-02-13 04:40:53 -08:00 · 2025-02-13 04:40:53 -08:00 · de7e76c250
commit de7e76c250
parent 1013ab5891
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/rl_replacements.py
+++ b/unsloth/models/rl_replacements.py
@ -108,6 +108,7 @@ def _sft_trainer_compute_loss(self, model, inputs, return_outputs = False, num_i
        return_outputs = return_outputs,
        num_items_in_batch = num_items_in_batch,
    )
+    print(loss, outputs)
    return (loss, outputs) if return_outputs else loss
 pass