plaguss
/

Mistral-7B-v0.1-Math-Shepherd-PRM-token-0.1

Token Classification

Generated from Trainer

stepwise-reward-trainer

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

plaguss HF staff commited on 30 days ago

Commit

d642a7a

•

1 Parent(s): e48a617

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-token-0.1
 tags:
@@ -11,7 +12,7 @@ licence: license
 # Model Card for Mistral-7B-v0.1-Math-Shepherd-PRM-token-0.1
-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: plaguss/math_shepherd_token
 library_name: transformers
 model_name: Mistral-7B-v0.1-Math-Shepherd-PRM-token-0.1
 tags:
 # Model Card for Mistral-7B-v0.1-Math-Shepherd-PRM-token-0.1
+This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the [plaguss/math_shepherd_token](https://huggingface.co/datasets/plaguss/math_shepherd_token) dataset.
 It has been trained using [TRL](https://github.com/huggingface/trl).
 ## Quick start