From ea0d11cd2ffeca872f8b77cc2236d461e8f1dfea Mon Sep 17 00:00:00 2001 From: Goekdeniz-Guelmez Date: Sun, 19 Jan 2025 02:05:43 +0100 Subject: [PATCH] update --- llms/mlx_lm/tuner/orpo_trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llms/mlx_lm/tuner/orpo_trainer.py b/llms/mlx_lm/tuner/orpo_trainer.py index 4eaace31..608d0dd0 100644 --- a/llms/mlx_lm/tuner/orpo_trainer.py +++ b/llms/mlx_lm/tuner/orpo_trainer.py @@ -84,6 +84,7 @@ def orpo_loss( return loss, reward, num_tokens + def evaluate_orpo( model, dataset,