Update README.md

README.md

@@ -2,6 +2,10 @@
 base_model: Qwen/Qwen2.5-Math-7B-Instruct
 library_name: transformers
 model_name: qwen-prm-7b-soft-labels
+datasets:
+- axolotl-ai-co/Math-Shepherd
+- axolotl-ai-co/prm800k_phase_1
+- axolotl-ai-co/prm800k_phase_2
 tags:
 - generated_from_trainer
 - axolotl
@@ -10,10 +14,12 @@ tags:
 licence: license
 ---
 
+[<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
+
 # Model Card for qwen-prm-7b-soft-labels
 
 This model is a fine-tuned version of [Qwen/Qwen2.5-Math-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Math-7B-Instruct).
-It has been trained using [TRL](https://github.com/huggingface/trl).
+It has been trained using [Axolotl](https://github.com/axolotl-ai-cloud/axolotl) with [TRL](https://github.com/huggingface/trl).
 
 ## Quick start