From a7273f6a56ea25bc77b30d8cee6f83b93d676ca8 Mon Sep 17 00:00:00 2001 From: Goekdeniz-Guelmez Date: Wed, 12 Feb 2025 18:30:12 +0100 Subject: [PATCH] small fix --- 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 fb19ba50..2ca012f3 100644 --- a/llms/mlx_lm/tuner/datasets.py +++ b/llms/mlx_lm/tuner/datasets.py @@ -310,7 +310,7 @@ def load_dataset(args, tokenizer: PreTrainedTokenizer): train, valid, test = load_local_dataset(args, data_path, tokenizer, args) else: print(f"Loading Hugging Face dataset {args.data}.") - train, valid, test = load_hf_dataset(args.data, tokenizer, args) + train, valid, test = load_hf_dataset(args, args.data, tokenizer, args) if args.train and len(train) == 0: raise ValueError(