winglian commited on
Commit
a27d594
·
1 Parent(s): 1fb0376

fix conditional so alpaca doesn't choke

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/data.py +1 -1
src/axolotl/utils/data.py CHANGED
@@ -86,7 +86,7 @@ def load_prepare_datasets(tokenizer, cfg, default_dataset_prepared_path):
86
  )
87
  ds_wrapper = TokenizedPromptDataset(ds_strategy, ds["train"])
88
  datasets.append(ds_wrapper)
89
- if d.type == "jeopardy":
90
  ds_strategy = JeopardyPromptTokenizingStrategy(
91
  JeopardyPrompter(), tokenizer, cfg.train_on_inputs, cfg.sequence_len
92
  )
 
86
  )
87
  ds_wrapper = TokenizedPromptDataset(ds_strategy, ds["train"])
88
  datasets.append(ds_wrapper)
89
+ elif d.type == "jeopardy":
90
  ds_strategy = JeopardyPromptTokenizingStrategy(
91
  JeopardyPrompter(), tokenizer, cfg.train_on_inputs, cfg.sequence_len
92
  )