plaguss
/

Mistral-7B-v0.1-Math-Shepherd-PRM-0.2

Token Classification

Generated from Trainer

stepwise-reward-trainer

text-generation-inference

Model card Files Files and versions

plaguss commited on Dec 10, 2024

Commit

9791a93

·

verified ·

1 Parent(s): 5d4da06

End of training

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -1,5 +1,6 @@
 ---
 base_model: mistralai/Mistral-7B-v0.1
 library_name: transformers
 model_name: Mistral-7B-v0.1-Math-Shepherd-PRM-0.2
 tags:
@@ -11,7 +12,7 @@ licence: license
 # Model Card for Mistral-7B-v0.1-Math-Shepherd-PRM-0.2
-This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1).
 It has been trained using [TRL](https://github.com/huggingface/trl).
 ## Quick start

 ---
 base_model: mistralai/Mistral-7B-v0.1
+datasets: trl-lib/math_shepherd
 library_name: transformers
 model_name: Mistral-7B-v0.1-Math-Shepherd-PRM-0.2
 tags:
 # Model Card for Mistral-7B-v0.1-Math-Shepherd-PRM-0.2
+This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the [trl-lib/math_shepherd](https://huggingface.co/datasets/trl-lib/math_shepherd) dataset.
 It has been trained using [TRL](https://github.com/huggingface/trl).
 ## Quick start