matsuo-lab
/

weblab-10b-instruction-sft

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Takeshi Kojima commited on Aug 4, 2023

Commit

0c62a4c

•

1 Parent(s): f60ee9c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -64,7 +64,7 @@ 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", use_fast=False)
 model = AutoModelForCausalLM.from_pretrained("Kojima777/weblab-10b-instruction-sft")
 if torch.cuda.is_available():

 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():