matsuo-lab
/

weblab-10b-instruction-sft

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

matsuo-lab commited on Aug 17, 2023

Commit

e9d3e3a

•

1 Parent(s): a4ec091

Update README.md

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -36,8 +36,8 @@ This repository provides a Japanese-centric multilingual GPT-NeoX model of 10 bi
     | Variant | Link |
     | :-- | :--|
-    | weblab-10b-instruction-sft | https://huggingface.co/Kojima777/weblab-10b-instruction-sft |
-    | weblab-10b | https://huggingface.co/Kojima777/weblab-10b |
 * **Authors**
@@ -67,8 +67,8 @@ This repository provides a Japanese-centric multilingual GPT-NeoX model of 10 bi
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-tokenizer = AutoTokenizer.from_pretrained("Kojima777/weblab-10b-instruction-sft")
-model = AutoModelForCausalLM.from_pretrained("Kojima777/weblab-10b-instruction-sft")
 if torch.cuda.is_available():
     model = model.to("cuda")

     | Variant | Link |
     | :-- | :--|
+    | weblab-10b-instruction-sft | https://huggingface.co/matsuo-lab/weblab-10b-instruction-sft |
+    | weblab-10b | https://huggingface.co/matsuo-lab/weblab-10b |
 * **Authors**
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("matsuo-lab/weblab-10b-instruction-sft")
+model = AutoModelForCausalLM.from_pretrained("matsuo-lab/weblab-10b-instruction-sft")
 if torch.cuda.is_available():
     model = model.to("cuda")