Update README.md

README.md

@@ -5,14 +5,17 @@ tags:
 - summarization
 - generated_from_trainer
 model-index:
-- name: cpo-sum-10p-shuff-1e-7-full-gemma
+- name: gemma-2-2b-Sum-CPO
   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. -->
 
-# cpo-sum-10p-shuff-1e-7-full-gemma
+# gemma-2-2b-Sum-CPO
 
 This model is a fine-tuned version of [google/gemma-2-2b](https://huggingface.co/google/gemma-2-2b) on the openai/summarize_from_feedback dataset.
 
@@ -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