matsuo-lab commited on
Commit
0e13cf6
1 Parent(s): c5e01d8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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/Kojima777/weblab-10b-instruction-sft |
30
- | weblab-10b | https://huggingface.co/Kojima777/weblab-10b |
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("Kojima777/weblab-10b")
61
- model = AutoModelForCausalLM.from_pretrained("Kojima777/weblab-10b")
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")