Update README.md

README.md

@@ -1,11 +1,14 @@
 ---
 license: apache-2.0
-base_model: plaguss/zephyr-7b-spin-iter2-v0
+base_model: argilla/zephyr-7b-spin-iter2-v0
 tags:
 - generated_from_trainer
 model-index:
 - name: zephyr-7b-spin-iter3-v0
   results: []
+datasets:
+- argilla/10k_prompts_SPIN_iter3_zephyr_top
+- argilla/10k_prompts_SPIN_iter2_zephyr_top
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -13,7 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # zephyr-7b-spin-iter3-v0
 
-This model is a fine-tuned version of [plaguss/zephyr-7b-spin-iter2-v0](https://huggingface.co/plaguss/zephyr-7b-spin-iter2-v0) on the None dataset.
+This model is a fine-tuned version of [argilla/zephyr-7b-spin-iter2-v0](https://huggingface.co/argilla/zephyr-7b-spin-iter2-v0) on the
+[argilla/10k_prompts_SPIN_iter3_zephyr_top](https://huggingface.co/datasets/argilla/10k_prompts_SPIN_iter3_zephyr_top) and the 
+[argilla/10k_prompts_SPIN_iter2_zephyr_top](https://huggingface.co/datasets/argilla/10k_prompts_SPIN_iter2_zephyr_top) dataset.
+
 It achieves the following results on the evaluation set:
 - Loss: 0.1099
 - Rewards/real: -2.9181
@@ -71,4 +77,4 @@ The following hyperparameters were used during training:
 - Transformers 4.37.0
 - Pytorch 2.1.2+cu121
 - Datasets 2.14.6
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2