[Model] Add Qwen2 PRM model support (#12202)

Signed-off-by: Isotr0py <2037008807@qq.com>
2025-01-20 14:59:46 +08:00
parent 0974c9bc5c
commit 83609791d2
5 changed files with 45 additions and 13 deletions
--- a/docs/source/models/supported_models.md
+++ b/docs/source/models/supported_models.md
@@ -470,6 +470,11 @@ of the whole prompt are extracted from the normalized hidden state corresponding
  - `Qwen/Qwen2.5-Math-RM-72B`, etc.
  - ✅︎
  - ✅︎
+* - `Qwen2ForProcessRewardModel`
+  - Qwen2-based
+  - `Qwen/Qwen2.5-Math-PRM-7B`, `Qwen/Qwen2.5-Math-PRM-72B`, etc.
+  - ✅︎
+  - ✅︎
 ```

 If your model is not in the above list, we will try to automatically convert the model using