Update README.md

README.md

@@ -7,13 +7,13 @@ tags:
 - causal-lm
 - small-lm
 datasets:
-- roneneldan/TinyStories
+- ltg/babylm-2024-baby-cosmo-fine-10m
 pipeline_tag: text-generation
 ---
 
 # TinyStories 30M - Base Model
 
-A 30M parameter causal language model pre-trained on the TinyStories dataset.
+A 30M parameter causal language model pre-trained on the ltg/babylm-2024-baby-cosmo-fine-10m
 
 ## Model Details
 
@@ -38,7 +38,7 @@ print(tokenizer.decode(outputs[0]))
 
 ## Training Details
 
-This is the base pre-trained model before instruction tuning. It was trained on the Babylm dataset which contains simple children's phrases
+This is the base pre-trained model before instruction tuning. It was trained on the ltg/babylm-2024-baby-cosmo-fine-10m dataset which contains simple children's phrases
 
 ## Related Models