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

2024-08-21 10:37:51 -07:00
parent 9f917245f6
commit f07802f9fa
1 changed files with 3 additions and 3 deletions
--- 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