Text Generation
PEFT
Safetensors
English
process-reward-model
prm
retrieval-augmented-generation
lora
conversational
Instructions to use MinKeonKim/PRO-STEP-PRM-8B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use MinKeonKim/PRO-STEP-PRM-8B with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-0528-Qwen3-8B") model = PeftModel.from_pretrained(base_model, "MinKeonKim/PRO-STEP-PRM-8B") - Notebooks
- Google Colab
- Kaggle
Add pipeline_tag and paper/GitHub links
Browse filesThis PR improves the model card by adding the missing `pipeline_tag: text-generation` to the YAML metadata and by adding direct links to the paper and GitHub code repository in the README. The existing metadata and content have been preserved.
README.md
CHANGED
|
@@ -1,20 +1,24 @@
|
|
| 1 |
---
|
| 2 |
-
license: mit
|
| 3 |
base_model: deepseek-ai/DeepSeek-R1-0528-Qwen3-8B
|
| 4 |
datasets:
|
| 5 |
-
|
| 6 |
language:
|
| 7 |
-
|
| 8 |
library_name: peft
|
|
|
|
|
|
|
| 9 |
tags:
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
---
|
| 15 |
|
| 16 |
# PRO-STEP: Process Reward Model for Agentic RAG
|
| 17 |
|
|
|
|
|
|
|
|
|
|
| 18 |
The supervised step-level **Process Reward Model (PRM)** for [PRO-STEP](https://huggingface.co/DORAEMONG/PRO-STEP-Policy-7B). Trained to evaluate agentic-RAG steps along six axes (entity grounding, search query quality, reasoning, answer specificity, recovery, overconfidence).
|
| 19 |
|
| 20 |
- **Base model**: DeepSeek-R1-0528-Qwen3-8B
|
|
@@ -58,4 +62,4 @@ tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-0528-Qwen3-8B
|
|
| 58 |
author={...},
|
| 59 |
year={2026}
|
| 60 |
}
|
| 61 |
-
```
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
base_model: deepseek-ai/DeepSeek-R1-0528-Qwen3-8B
|
| 3 |
datasets:
|
| 4 |
+
- DORAEMONG/PRO-STEP-PRM-Data
|
| 5 |
language:
|
| 6 |
+
- en
|
| 7 |
library_name: peft
|
| 8 |
+
license: mit
|
| 9 |
+
pipeline_tag: text-generation
|
| 10 |
tags:
|
| 11 |
+
- process-reward-model
|
| 12 |
+
- prm
|
| 13 |
+
- retrieval-augmented-generation
|
| 14 |
+
- lora
|
| 15 |
---
|
| 16 |
|
| 17 |
# PRO-STEP: Process Reward Model for Agentic RAG
|
| 18 |
|
| 19 |
+
**Paper**: [PRO-STEP: Step-level Process Reward Optimization for Retrieval-Augmented Generation](https://huggingface.co/papers/2609.01658)
|
| 20 |
+
**Code**: [github.com/keemminnke/PRO-Step](https://github.com/keemminnke/PRO-Step)
|
| 21 |
+
|
| 22 |
The supervised step-level **Process Reward Model (PRM)** for [PRO-STEP](https://huggingface.co/DORAEMONG/PRO-STEP-Policy-7B). Trained to evaluate agentic-RAG steps along six axes (entity grounding, search query quality, reasoning, answer specificity, recovery, overconfidence).
|
| 23 |
|
| 24 |
- **Base model**: DeepSeek-R1-0528-Qwen3-8B
|
|
|
|
| 62 |
author={...},
|
| 63 |
year={2026}
|
| 64 |
}
|
| 65 |
+
```
|