Update README.md

README.md

@@ -24,15 +24,13 @@ This repository provides a small-sized Japanese GPT-2 model. The model was train
 
 # How to use the model
 
-*NOTE:* Use `T5Tokenizer` to initiate the tokenizer.
-
 ~~~~
-from transformers import T5Tokenizer, GPT2LMHeadModel
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = T5Tokenizer.from_pretrained("rinna/japanese-gpt2-small")
+tokenizer = AutoTokenizer.from_pretrained("rinna/japanese-gpt2-small", use_fast=False)
 tokenizer.do_lower_case = True  # due to some bug of tokenizer config loading
 
-model = GPT2LMHeadModel.from_pretrained("rinna/japanese-gpt2-small")
+model = AutoModelForCausalLM.from_pretrained("rinna/japanese-gpt2-small")
 ~~~~
 
 # Model architecture