kajyuuen alfredplpl commited on
Commit
336f9e0
1 Parent(s): 14bd5e4

Update README.md (#2)

Browse files

- Update README.md (6526b23e06083b2e14f75b164374f7cde5ea4278)


Co-authored-by: Yasunori Ozaki <alfredplpl@users.noreply.huggingface.co>

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