matsuo-lab
/

weblab-10b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Takeshi Kojima commited on Aug 4, 2023

Commit

a1205ba

·

1 Parent(s): 98be71c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

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

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("Kojima777/weblab-10b")
 model = AutoModelForCausalLM.from_pretrained("Kojima777/weblab-10b")
 if torch.cuda.is_available():