alfredplpl commited on
Commit
6526b23
1 Parent(s): 14bd5e4

Update README.md

Browse files

AutoModelForCausalLM is not AutoModel.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -19,7 +19,7 @@ import torch
19
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
20
 
21
  tokenizer = AutoTokenizer.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft", use_fast=False)
22
- model = AutoModel.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft")
23
 
24
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
25
 
 
19
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
20
 
21
  tokenizer = AutoTokenizer.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft", use_fast=False)
22
+ model = AutoModelForCausalLM.from_pretrained("line-corporation/japanese-large-lm-3.6b-instruction-sft")
23
 
24
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
25