Update README.md
Browse files
README.md
CHANGED
@@ -18,7 +18,7 @@ Usage:
|
|
18 |
import transformers
|
19 |
|
20 |
model = transformers.AutoModelForCausalLM.from_pretrained("CarperAI/7b-alpha")
|
21 |
-
tokenizer = transformers.AutoTokenizer.from_pretrained("
|
22 |
tokenizer.pad_token = tokenizer.eos_token
|
23 |
tokenizer.paddding_side = "left"
|
24 |
|
|
|
18 |
import transformers
|
19 |
|
20 |
model = transformers.AutoModelForCausalLM.from_pretrained("CarperAI/7b-alpha")
|
21 |
+
tokenizer = transformers.AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-20b")
|
22 |
tokenizer.pad_token = tokenizer.eos_token
|
23 |
tokenizer.paddding_side = "left"
|
24 |
|