alfredplpl commited on
Commit
703614e
1 Parent(s): 14bd5e4

Update README.md

Browse files

We need to import AutoModel class.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ This repository provides a 3.6B parameters Japanese language model, fine-tuned a
16
 
17
  ```python
18
  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")
 
16
 
17
  ```python
18
  import torch
19
+ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, AutoModel
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")