line-corporation
/

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

Text Generation

text-generation-inference

Model card Files Files and versions Community

Update README.md

#2

by alfredplpl - opened Aug 18, 2023

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

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)