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

Fix keyword argument invokation

Browse Source
This commit is contained in:
Chime Ogbuji 2024-11-03 20:26:15 -05:00 committed by Awni Hannun
parent 387c45efa2
commit 78c33e5037
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -211,7 +211,6 @@ def load_custom_hf_dataset(args, tokenizer: PreTrainedTokenizer):
prompt_f = hf_args.get("prompt_feature", None) prompt_f = hf_args.get("prompt_feature", None)
completion_f = hf_args.get("completion_feature", None) completion_f = hf_args.get("completion_feature", None)
chat_f = hf_args.get("chat_feature", None) chat_f = hf_args.get("chat_feature", None)
if args.train: if args.train:
train_split = hf_args.get("train_split", "train[:80%]") train_split = hf_args.get("train_split", "train[:80%]")
valid_split = hf_args.get("valid_split", "train[-10%:]") valid_split = hf_args.get("valid_split", "train[-10%:]")