From df1406735b1e140e44b29b302783fbc837da9269 Mon Sep 17 00:00:00 2001 From: Victor Nogueira Date: Tue, 21 Jan 2025 23:12:43 +0100 Subject: [PATCH] Fix dataset variable name, in `datasets.py` (#1212) --- 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 "