Update README.md

README.md

@@ -1,20 +1,23 @@
 ---
 license: gemma
-base_model: martimfasantos/sft-sum-chosen-10lp-shuff-full-gemma
+base_model: martimfasantos/gemma-2-2b-Sum-SFT
 tags:
 - summarization
 - generated_from_trainer
 model-index:
-- name: dpo-sum-10p-shuff-1e-7-full-from-sft-gemma
+- name: gemma-2-2b-Sum-DPO
   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
 should probably proofread and complete it, then remove this comment. -->
 
-# dpo-sum-10p-shuff-1e-7-full-from-sft-gemma
+# gemma-2-2b-Sum-DPO
 
-This model is a fine-tuned version of [martimfasantos/sft-sum-chosen-10lp-shuff-full-gemma](https://huggingface.co/martimfasantos/sft-sum-chosen-10lp-shuff-full-gemma) on the openai/summarize_from_feedback dataset.
+This model is a fine-tuned version of [martimfasantos/gemma-2-2b-Sum-SFT](https://huggingface.co/martimfasantos/gemma-2-2b-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