Update README.md

README.md

@@ -1,12 +1,15 @@
 ---
 license: apache-2.0
-base_model: martimfasantos/sft-sum-chosen-10lp-shuff-full-tiny_same_params
+base_model: martimfasantos/TinyLlama-1.1B-Sum-SFT
 tags:
 - summarization
 - generated_from_trainer
 model-index:
-- name: simpo-sum-10p-shuff-1e-7-full-from-sft-tiny_real
+- name: TinyLlama-1.1B-Sum-DPO-gamma
   results: []
+datasets:
+- openai/summarize_from_feedback
+pipeline_tag: summarization
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -14,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # simpo-sum-10p-shuff-1e-7-full-from-sft-tiny_real
 
-This model is a fine-tuned version of [martimfasantos/sft-sum-chosen-10lp-shuff-full-tiny_same_params](https://huggingface.co/martimfasantos/sft-sum-chosen-10lp-shuff-full-tiny_same_params) on the openai/summarize_from_feedback dataset.
+This model is a fine-tuned version of [martimfasantos/TinyLlama-1.1B-Sum-SFT](https://huggingface.co/martimfasantos/TinyLlama-1.1B-Sum-SFT) on the openai/summarize_from_feedback dataset.
 
 ## Model description
 
@@ -56,4 +59,4 @@ The following hyperparameters were used during training:
 - Transformers 4.43.3
 - Pytorch 2.3.1+cu121
 - Datasets 2.20.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1