line-corporation
/

japanese-large-lm-3.6b-instruction-sft

Text Generation

text-generation-inference

Model card Files Files and versions Community

alfredplpl commited on Aug 18, 2023

Commit

6526b23

•

1 Parent(s): 14bd5e4

Update README.md

AutoModelForCausalLM is not AutoModel.

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 tokenizer = AutoTokenizer.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft", use_fast=False)
-model = AutoModel.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)

 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 tokenizer = AutoTokenizer.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft", use_fast=False)
+model = AutoModelForCausalLM.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)