Takeshi Kojima
commited on
Commit
•
0c62a4c
1
Parent(s):
f60ee9c
Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ This repository provides a Japanese-centric multilingual GPT-NeoX model of 10 bi
|
|
64 |
import torch
|
65 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
66 |
|
67 |
-
tokenizer = AutoTokenizer.from_pretrained("Kojima777/weblab-10b-instruction-sft"
|
68 |
model = AutoModelForCausalLM.from_pretrained("Kojima777/weblab-10b-instruction-sft")
|
69 |
|
70 |
if torch.cuda.is_available():
|
|
|
64 |
import torch
|
65 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
66 |
|
67 |
+
tokenizer = AutoTokenizer.from_pretrained("Kojima777/weblab-10b-instruction-sft")
|
68 |
model = AutoModelForCausalLM.from_pretrained("Kojima777/weblab-10b-instruction-sft")
|
69 |
|
70 |
if torch.cuda.is_available():
|