Update README.md
Browse files
README.md
CHANGED
|
@@ -44,8 +44,8 @@ Additionally, the model demonstrates strong generalization to undergraduate-leve
|
|
| 44 |
```python
|
| 45 |
# Example code for loading and using the tactic generator model
|
| 46 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 47 |
-
model = AutoModelForCausalLM.from_pretrained("ByteDance-Seed/BFS-Prover-V2-
|
| 48 |
-
tokenizer = AutoTokenizer.from_pretrained("ByteDance-Seed/BFS-Prover-V2-
|
| 49 |
|
| 50 |
# imo_1964_p2 from miniF2F
|
| 51 |
state = """a b c : ℝ
|
|
@@ -68,7 +68,7 @@ outputs = model.generate(**inputs)
|
|
| 68 |
tactic = tokenizer.decode(outputs[0], skip_special_tokens=True).split(sep)[1]
|
| 69 |
print(tactic)
|
| 70 |
|
| 71 |
-
# Generated tactic: "nlinarith [sq_nonneg (
|
| 72 |
```
|
| 73 |
|
| 74 |
## 📚 Citation
|
|
|
|
| 44 |
```python
|
| 45 |
# Example code for loading and using the tactic generator model
|
| 46 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 47 |
+
model = AutoModelForCausalLM.from_pretrained("ByteDance-Seed/BFS-Prover-V2-7B")
|
| 48 |
+
tokenizer = AutoTokenizer.from_pretrained("ByteDance-Seed/BFS-Prover-V2-7B")
|
| 49 |
|
| 50 |
# imo_1964_p2 from miniF2F
|
| 51 |
state = """a b c : ℝ
|
|
|
|
| 68 |
tactic = tokenizer.decode(outputs[0], skip_special_tokens=True).split(sep)[1]
|
| 69 |
print(tactic)
|
| 70 |
|
| 71 |
+
# Generated tactic: "nlinarith [sq_nonneg (c - a), sq_nonneg (b - a), sq_nonneg (c - b), h₀.1, h₀.2.1, h₀.2.2, h₁.le, h₂.le, h₃.le]"
|
| 72 |
```
|
| 73 |
|
| 74 |
## 📚 Citation
|