Update README.md

README.md

@@ -12,8 +12,8 @@ This model is a 13.5 billion parameter model that combines four sets of 16 layer
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-tokenizer = AutoTokenizer.from_pretrained("sudy-super/baku-13b")
-model = AutoModelForCausalLM.from_pretrained("sudy-super/baku-13b", device_map="auto", torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained("sudy-super/baku-13b-v2")
+model = AutoModelForCausalLM.from_pretrained("sudy-super/baku-13b-v2", device_map="auto", torch_dtype=torch.bfloat16)
 prompt = "大規模言語モデルとは、"
 token_ids = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
 with torch.no_grad():