plaguss
/

Qwen2.5-0.5B-Math-Shepherd-PRM-0.2

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 13 days ago

Commit

32dbfed

•

1 Parent(s): 1bbb401

End of training

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

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

 ---
 base_model: Qwen/Qwen2.5-0.5B
+datasets: trl-lib/math_shepherd
 library_name: transformers
 model_name: Qwen2.5-0.5B-Math-Shepherd-PRM-0.2
 tags:
 # Model Card for Qwen2.5-0.5B-Math-Shepherd-PRM-0.2
+This model is a fine-tuned version of [Qwen/Qwen2.5-0.5B](https://huggingface.co/Qwen/Qwen2.5-0.5B) 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