nits

2025-09-19 03:18:06 +08:00 · 2025-01-24 18:13:05 +01:00
parent 54fcd8ed63
commit 531c3345c6
1 changed files with 1 additions and 1 deletions
--- 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 = []