fix dict check

OpenAccess-AI-Collective · Jan 5, 2024 · 02dc87f · 02dc87f
1 parent 871355a
commit 02dc87f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/axolotl/utils/data.py b/src/axolotl/utils/data.py
@@ -71,9 +71,9 @@ def prepare_dataset(cfg, tokenizer):
  else:
  path = cfg.pretraining_dataset
  name = None
- if isinstance(dict, cfg.pretraining_dataset):
- path = cfg.pretraining_dataset.path
- name = cfg.pretraining_dataset.name
+ if isinstance(cfg.pretraining_dataset, dict):
+ path = cfg.pretraining_dataset["path"]
+ name = cfg.pretraining_dataset["name"]
 
  train_dataset = load_pretraining_dataset(
  path,