Update README.md

README.md

@@ -7,6 +7,7 @@ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T
 model-index:
 - name: airoboros-lora-out
   results: []
+pipeline_tag: text-generation
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
 # airoboros-lora-out
 
-This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T) on the None dataset.
+This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T) on the `jondurbin/airoboros-3.1` dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7230
 
@@ -29,7 +30,7 @@ More information needed
 
 ## Training and evaluation data
 
-More information needed
+https://wandb.ai/wing-lian/airoboros-tinyllama
 
 ## Training procedure
 
@@ -81,4 +82,4 @@ The following `bitsandbytes` quantization config was used during training:
 ### Framework versions
 
 
-- PEFT 0.6.0
+- PEFT 0.6.0