fix dict check

axolotl-ai-cloud · 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,