From 531c3345c648a39f704a1f0a8cf26a01aa63e00d Mon Sep 17 00:00:00 2001 From: Goekdeniz-Guelmez Date: Fri, 24 Jan 2025 18:13:05 +0100 Subject: [PATCH] nits --- llms/mlx_lm/tuner/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/mlx_lm/tuner/datasets.py b/llms/mlx_lm/tuner/datasets.py index 0f23c2a4..98f01ef3 100644 --- a/llms/mlx_lm/tuner/datasets.py +++ b/llms/mlx_lm/tuner/datasets.py @@ -19,7 +19,7 @@ class DPODataset: prompt_key: str = "prompt", chosen_key: str = "chosen", rejected_key: str = "rejected", - system_key: str = None + system_key: str = "system", ): self._chosen_data = []