Add license and pipeline tag (#1)
Browse files- Add license and pipeline tag (4e7d4400728549301072dc3d01c9091b2638dddb)
Co-authored-by: Niels Rogge <nielsr@users.noreply.huggingface.co>
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(
|