Update README.md
Browse files
README.md
CHANGED
@@ -7,4 +7,11 @@ widget:
|
|
7 |
---
|
8 |
|
9 |
This model was created using GPT-2 as a base, and fine-tuned upon a dataset of elementary school problems requiring logic and reasoning.
|
10 |
-
Requires Pytorch
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
---
|
8 |
|
9 |
This model was created using GPT-2 as a base, and fine-tuned upon a dataset of elementary school problems requiring logic and reasoning.
|
10 |
+
Requires Pytorch
|
11 |
+
|
12 |
+
How to use to infer text
|
13 |
+
```python
|
14 |
+
type = "gpt2-large"
|
15 |
+
tokenizer = AutoTokenizer.from_pretrained(type)
|
16 |
+
model = AutoModelForCausalLM.from_pretrained(type)
|
17 |
+
```
|