Update README.md

README.md

@@ -22,8 +22,8 @@ language:
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("lightblue/karasu-7B-chat")
-model = AutoModelForCausalLM.from_pretrained("lightblue/karasu-7B-chat", torch_dtype=torch.bfloat16, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained("lightblue/karasu-7B-chat-plus")
+model = AutoModelForCausalLM.from_pretrained("lightblue/karasu-7B-chat-plus", torch_dtype=torch.bfloat16, device_map="auto")
 
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
@@ -41,7 +41,7 @@ pipe(prompt, max_new_tokens=100, do_sample=False, temperature=0.0, return_full_t
 from vllm import LLM, SamplingParams
 
 sampling_params = SamplingParams(temperature=0.0, max_tokens=100)
-llm = LLM(model="lightblue/karasu-7B-chat")
+llm = LLM(model="lightblue/karasu-7B-chat-plus")
 
 messages = [{"role": "system", "content": "あなたはAIアシスタントです。"}]
 messages.append({"role": "user", "content": "イギリスの首相は誰ですか？"})