abeja
/

gpt-neox-japanese-2.7b

Text Generation

gpt_neox_japanese

Inference Endpoints

Model card Files Files and versions Community

SO0529 commited on Sep 20, 2022

Commit

09291e0

•

1 Parent(s): 56d755d

fix typo

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -65,7 +65,7 @@ model = AutoModelForCausalLM.from_pretrained("abeja/gpt-neox-japanese-2.7b")
 input_text = "人とAIが協調するためには、"
 input_ids = tokenizer.encode(input_text, return_tensors="pt")
-gen_tokens = lm_model.generate(
     input_ids,
     max_length=100,
     do_sample=True,

 input_text = "人とAIが協調するためには、"
 input_ids = tokenizer.encode(input_text, return_tensors="pt")
+gen_tokens = model.generate(
     input_ids,
     max_length=100,
     do_sample=True,