From 78c33e503794c5285dda1bc335cbc670b1197d99 Mon Sep 17 00:00:00 2001 From: Chime Ogbuji Date: Sun, 3 Nov 2024 20:26:15 -0500 Subject: [PATCH] Fix keyword argument invokation --- llms/mlx_lm/tuner/datasets.py | 1 - 1 file changed, 1 deletion(-) diff --git a/llms/mlx_lm/tuner/datasets.py b/llms/mlx_lm/tuner/datasets.py index 5e4e7ece..4599772e 100644 --- a/llms/mlx_lm/tuner/datasets.py +++ b/llms/mlx_lm/tuner/datasets.py @@ -211,7 +211,6 @@ def load_custom_hf_dataset(args, tokenizer: PreTrainedTokenizer): prompt_f = hf_args.get("prompt_feature", None) completion_f = hf_args.get("completion_feature", None) chat_f = hf_args.get("chat_feature", None) - if args.train: train_split = hf_args.get("train_split", "train[:80%]") valid_split = hf_args.get("valid_split", "train[-10%:]")