alfredplpl
commited on
Commit
•
703614e
1
Parent(s):
14bd5e4
Update README.md
Browse filesWe need to import AutoModel class.
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")
|