examples: fix tiny-llama pretrain yml syntax (#1840)

axolotl-ai-cloud · Aug 21, 2024 · f07802f · f07802f
1 parent 9f91724
commit f07802f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/tiny-llama/pretrain.yml b/examples/tiny-llama/pretrain.yml
@@ -9,9 +9,9 @@ strict: false
 
 max_steps: 200
 pretraining_dataset:
-  path: c4
-  name: en
-  type: pretrain
+  - path: allenai/c4
+    name: en
+    type: pretrain
 dataset_prepared_path:
 val_set_size: 0.0
 output_dir: ./outputs/model-out