fix model name
Browse files
README.md
CHANGED
@@ -64,8 +64,8 @@ print(prompt)
|
|
64 |
import torch
|
65 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
66 |
|
67 |
-
tokenizer = AutoTokenizer.from_pretrained(".", use_fast=False)
|
68 |
-
model = AutoModelForCausalLM.from_pretrained(".")
|
69 |
|
70 |
if torch.cuda.is_available():
|
71 |
model = model.to("cuda")
|
|
|
64 |
import torch
|
65 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
66 |
|
67 |
+
tokenizer = AutoTokenizer.from_pretrained("rinna/japanese-gpt-neox-3.6b-instruction-sft", use_fast=False)
|
68 |
+
model = AutoModelForCausalLM.from_pretrained("rinna/japanese-gpt-neox-3.6b-instruction-sft")
|
69 |
|
70 |
if torch.cuda.is_available():
|
71 |
model = model.to("cuda")
|