Update README.md
Browse files
README.md
CHANGED
@@ -4,11 +4,16 @@ Text Generation Informal Formal
|
|
4 |
|
5 |
Trained on this model: https://huggingface.co/CarperAI/FIM-NeoX-1.3B, which is geared toward filling in the blank. Check out their model and give them a like!
|
6 |
|
|
|
|
|
|
|
|
|
|
|
7 |
```
|
8 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
9 |
|
10 |
-
|
11 |
-
|
|
|
12 |
```
|
13 |
|
14 |
```
|
|
|
4 |
|
5 |
Trained on this model: https://huggingface.co/CarperAI/FIM-NeoX-1.3B, which is geared toward filling in the blank. Check out their model and give them a like!
|
6 |
|
7 |
+
|
8 |
+
```
|
9 |
+
from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast
|
10 |
+
tokenizer = GPTNeoXTokenizerFast.from_pretrained("CarperAI/FIM-NeoX-1.3B")
|
11 |
+
model = GPTNeoXForCausalLM.from_pretrained("BigSalmon/FormalInformalConcise-FIM-NeoX-1.3B")
|
12 |
```
|
|
|
13 |
|
14 |
+
To load model, you may need to do:
|
15 |
+
```
|
16 |
+
pip install git+https://github.com/huggingface/transformers
|
17 |
```
|
18 |
|
19 |
```
|