From ff0372103f5828dd3b5f783b61d1f1f75d39f44f Mon Sep 17 00:00:00 2001 From: Victor Nogueira Date: Mon, 20 Jan 2025 11:36:20 +0100 Subject: [PATCH] Fix dataset variable name, in `datasets.py` --- 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 1b09c7e2..377e7cae 100644 --- a/llms/mlx_lm/tuner/datasets.py +++ b/llms/mlx_lm/tuner/datasets.py @@ -170,7 +170,7 @@ def load_custom_hf_dataset(args, tokenizer: PreTrainedTokenizer): if prompt_feature and completion_feature: return CompletionsDataset(ds, tokenizer, prompt_feature, completion_feature) elif text_feature: - return Dataset(train_ds, tokenizer, text_key=text_feature) + return Dataset(ds, tokenizer, text_key=text_feature) else: raise ValueError( "Specify either a prompt and completion feature or a text "