Update README.md

README.md

@@ -6,6 +6,12 @@ tags:
 model-index:
 - name: vicuna_1b_stage1
   results: []
+datasets:
+- Aeala/ShareGPT_Vicuna_unfiltered
+language:
+- en
+metrics:
+- accuracy
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,7 +21,7 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/momorami-kaist/medusa_test/runs/ku8hga4l)
 # vicuna_1b_stage1
 
-This model is a fine-tuned version of [Jiayi-Pan/Tiny-Vicuna-1B](https://huggingface.co/Jiayi-Pan/Tiny-Vicuna-1B) on the None dataset.
+This model is a fine-tuned version of [Jiayi-Pan/Tiny-Vicuna-1B](https://huggingface.co/Jiayi-Pan/Tiny-Vicuna-1B) on the Aeala/ShareGPT_Vicuna_unfiltered dataset.
 It achieves the following results on the evaluation set:
 - Loss: 2.9673
 
@@ -68,4 +74,4 @@ The following hyperparameters were used during training:
 - Transformers 4.43.0
 - Pytorch 2.3.1
 - Datasets 2.15.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1