Update README.md
Browse files
README.md
CHANGED
|
@@ -26,4 +26,27 @@ model_inputs = [get_prompt(text)]
|
|
| 26 |
|
| 27 |
print(model_inputs[0])
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
|
|
|
| 26 |
|
| 27 |
print(model_inputs[0])
|
| 28 |
|
| 29 |
+
### Example output :
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
<theorem>
|
| 33 |
+
theorem mathd_numbertheory_3 : (∑ x in Finset.range 10, (x + 1) ^ 2) % 10 = 5 := by
|
| 34 |
+
</theorem>
|
| 35 |
+
<conjecture>
|
| 36 |
+
theorem mathd_numbertheory_24 : ∑ k in Finset.range 11, k ^ 2 = 385 := by
|
| 37 |
+
</conjecture>
|
| 38 |
+
|
| 39 |
+
### Inference
|
| 40 |
+
model = LLM(model=model_name, seed=1, trust_remote_code=True, swap_space=8, tensor_parallel_size=1, max_model_len=4096)
|
| 41 |
+
|
| 42 |
+
sampling_params = SamplingParams(
|
| 43 |
+
temperature=1,
|
| 44 |
+
max_tokens=2048,
|
| 45 |
+
top_p=0.95,
|
| 46 |
+
n=1,
|
| 47 |
+
)
|
| 48 |
+
|
| 49 |
+
model_outputs = model.generate(model_inputs, sampling_params, use_tqdm=True)
|
| 50 |
+
print(model_outputs[0].outputs[0].text)
|
| 51 |
+
|
| 52 |
|