Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,8 @@ GPT-J 6B was pretrained on the [Pile](pile.eleuther.ai), a large scale curated d
|
|
43 |
|
44 |
### How to use
|
45 |
|
46 |
-
```
|
|
|
47 |
import torch
|
48 |
|
49 |
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
|
|
|
43 |
|
44 |
### How to use
|
45 |
|
46 |
+
```
|
47 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
48 |
import torch
|
49 |
|
50 |
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
|