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>
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 =
|
| 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 |
|