Add license and pipeline tag (#1)
Browse files- Add license and pipeline tag (4e7d4400728549301072dc3d01c9091b2638dddb)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -9,6 +9,8 @@ tags:
|
|
| 9 |
- verification
|
| 10 |
- math reasoning
|
| 11 |
- code verification
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for ThinkPRM-14B
|
|
@@ -60,7 +62,8 @@ llm = LLM(model=model_id, max_model_len=16384)
|
|
| 60 |
|
| 61 |
# Example problem and solution
|
| 62 |
problem = "Solve for x: 2x + 3 = 7"
|
| 63 |
-
prefix = "Step 1: Subtract 3 from both sides: 2x = 4
|
|
|
|
| 64 |
|
| 65 |
# Format the prompt
|
| 66 |
prompt = f"""You are given a math problem and a proposed step-by-step solution:
|
|
@@ -78,7 +81,8 @@ Review and critique each step in the proposed solution to determine whether each
|
|
| 78 |
|
| 79 |
prompt = tokenizer.apply_chat_template([
|
| 80 |
{'role': "user", "content": prompt}
|
| 81 |
-
], tokenize=False, add_generation_prompt=True) + "
|
|
|
|
| 82 |
|
| 83 |
# Set sampling parameters
|
| 84 |
sampling_params = SamplingParams(
|
|
|
|
| 9 |
- verification
|
| 10 |
- math reasoning
|
| 11 |
- code verification
|
| 12 |
+
license: apache-2.0
|
| 13 |
+
pipeline_tag: text-generation
|
| 14 |
---
|
| 15 |
|
| 16 |
# Model Card for ThinkPRM-14B
|
|
|
|
| 62 |
|
| 63 |
# Example problem and solution
|
| 64 |
problem = "Solve for x: 2x + 3 = 7"
|
| 65 |
+
prefix = "Step 1: Subtract 3 from both sides: 2x = 4
|
| 66 |
+
Step 2: Divide by 2: x = 1"
|
| 67 |
|
| 68 |
# Format the prompt
|
| 69 |
prompt = f"""You are given a math problem and a proposed step-by-step solution:
|
|
|
|
| 81 |
|
| 82 |
prompt = tokenizer.apply_chat_template([
|
| 83 |
{'role': "user", "content": prompt}
|
| 84 |
+
], tokenize=False, add_generation_prompt=True) + "
|
| 85 |
+
Let's verify step by step:"
|
| 86 |
|
| 87 |
# Set sampling parameters
|
| 88 |
sampling_params = SamplingParams(
|