From a27d594788764e095b7751f4eb5a109655b5f0e6 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 9 May 2023 20:57:07 -0400 Subject: [PATCH] fix conditional so alpaca doesn't choke --- src/axolotl/utils/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/data.py b/src/axolotl/utils/data.py index a6cc6cdcd..b217b50d7 100644 --- a/src/axolotl/utils/data.py +++ b/src/axolotl/utils/data.py @@ -86,7 +86,7 @@ def load_prepare_datasets(tokenizer, cfg, default_dataset_prepared_path): ) ds_wrapper = TokenizedPromptDataset(ds_strategy, ds["train"]) datasets.append(ds_wrapper) - if d.type == "jeopardy": + elif d.type == "jeopardy": ds_strategy = JeopardyPromptTokenizingStrategy( JeopardyPrompter(), tokenizer, cfg.train_on_inputs, cfg.sequence_len )