tianyuz commited on
Commit
e1d570e
1 Parent(s): 3d06736

fix model name

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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")