From 64e9824d3e475ddc4e253651025fdf969fb61c0e Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 21 Aug 2023 04:00:22 -0400 Subject: [PATCH] fix the data file --- src/axolotl/utils/callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/callbacks.py b/src/axolotl/utils/callbacks.py index 6fa809807..4d100a4be 100644 --- a/src/axolotl/utils/callbacks.py +++ b/src/axolotl/utils/callbacks.py @@ -125,7 +125,7 @@ def mmlu_eval_callback_factory(trainer, tokenizer): mmlu_dataset = load_dataset( "pharaouk/dharma-1", data_files={ - "eval": "pharaouk/dharma-1", + "eval": "dharma_eval.json", }, ) # mmlu_dataset = mmlu_dataset.remove_columns("subject")