zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-28 20:25:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

nits

Browse Source
This commit is contained in:
Goekdeniz-Guelmez 2025-01-24 18:13:05 +01:00
parent 54fcd8ed63
commit 531c3345c6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/tuner/datasets.py
View File

@ -19,7 +19,7 @@ class DPODataset:
prompt_key: str = "prompt", prompt_key: str = "prompt",
chosen_key: str = "chosen", chosen_key: str = "chosen",
rejected_key: str = "rejected", rejected_key: str = "rejected",
system_key: str = None system_key: str = "system",
): ):
self._chosen_data = [] self._chosen_data = []