Update README.md

README.md

@@ -15,11 +15,13 @@ inference:
     eta_cutoff: 0.001
     renormalize_logits: true
 widget:
-  - text: avocado chair
-    example_title: avocado chair
-  - text: A mysterious potato
-    example_title: potato
+- text: avocado chair
+  example_title: avocado chair
+- text: A mysterious potato
+  example_title: potato
 pipeline_tag: text-generation
+datasets:
+- pszemraj/midjourney-messages-cleaned
 ---
 
 # smol_llama-101M-midjourney-messages
@@ -51,4 +53,4 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.95) and epsilon=1e-08
 - lr_scheduler_type: inverse_sqrt
 - lr_scheduler_warmup_ratio: 0.05
-- num_epochs: 1.0
+- num_epochs: 1.0