matsuo-lab
commited on
Commit
•
0e13cf6
1
Parent(s):
c5e01d8
Update README.md
Browse files
README.md
CHANGED
@@ -26,8 +26,8 @@ This repository provides a Japanese-centric multilingual GPT-NeoX model of 10 bi
|
|
26 |
|
27 |
| Variant | Link |
|
28 |
| :-- | :--|
|
29 |
-
| weblab-10b-instruction-sft | https://huggingface.co/
|
30 |
-
| weblab-10b | https://huggingface.co/
|
31 |
|
32 |
* **Authors**
|
33 |
|
@@ -57,8 +57,8 @@ This repository provides a Japanese-centric multilingual GPT-NeoX model of 10 bi
|
|
57 |
import torch
|
58 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
59 |
|
60 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
61 |
-
model = AutoModelForCausalLM.from_pretrained("
|
62 |
|
63 |
if torch.cuda.is_available():
|
64 |
model = model.to("cuda")
|
|
|
26 |
|
27 |
| Variant | Link |
|
28 |
| :-- | :--|
|
29 |
+
| weblab-10b-instruction-sft | https://huggingface.co/matsuo-lab/weblab-10b-instruction-sft |
|
30 |
+
| weblab-10b | https://huggingface.co/matsuo-lab/weblab-10b |
|
31 |
|
32 |
* **Authors**
|
33 |
|
|
|
57 |
import torch
|
58 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
59 |
|
60 |
+
tokenizer = AutoTokenizer.from_pretrained("matsuo-lab/weblab-10b")
|
61 |
+
model = AutoModelForCausalLM.from_pretrained("matsuo-lab/weblab-10b")
|
62 |
|
63 |
if torch.cuda.is_available():
|
64 |
model = model.to("cuda")
|