Use tinystories in pretrain/debug.yaml (Lightning-AI#1203)

ugurcanvurgun · Mar 27, 2024 · a7acf99 · a7acf99
1 parent af8a39d
commit a7acf99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config_hub/pretrain/debug.yaml b/config_hub/pretrain/debug.yaml
@@ -20,7 +20,7 @@ initial_checkpoint_dir:
 resume: false
 
 # Data-related arguments. If not provided, the default is ``litgpt.data.TinyLlama``.
-data: OpenWebText
+data: TinyStories
 
 # Training-related arguments. See ``litgpt.args.TrainArgs`` for details
 train: