Commit
•
950bc98
1
Parent(s):
0762494
Fix example
Browse files
README.md
CHANGED
|
@@ -16,10 +16,11 @@ Works well with simple calculations, but fails with complex ones.
|
|
| 16 |
# Usage
|
| 17 |
|
| 18 |
```python
|
|
|
|
| 19 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 20 |
|
| 21 |
-
tokenizer = AutoTokenizer.from_pretrained("georgiyozhegov/calculator")
|
| 22 |
-
model = AutoModelForCausalLM.from_pretrained("georgiyozhegov/calculator")
|
| 23 |
|
| 24 |
prompt = "find 2 + 3\nstep"
|
| 25 |
|
|
|
|
| 16 |
# Usage
|
| 17 |
|
| 18 |
```python
|
| 19 |
+
import torch
|
| 20 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 21 |
|
| 22 |
+
tokenizer = AutoTokenizer.from_pretrained("georgiyozhegov/calculator-6m")
|
| 23 |
+
model = AutoModelForCausalLM.from_pretrained("georgiyozhegov/calculator-6m")
|
| 24 |
|
| 25 |
prompt = "find 2 + 3\nstep"
|
| 26 |
|