Update README.md

README.md

@@ -24,8 +24,8 @@ Qwen/Qwen-14B-Chat + Karasu's finetuning datasets
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-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")
+tokenizer = AutoTokenizer.from_pretrained("lightblue/qarasu-14B-chat-plus-unleashed", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("lightblue/qarasu-14B-chat-plus-unleashed", torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)
 
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
@@ -43,7 +43,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-plus")
+llm = LLM(model="lightblue/karasu-7B-chat-plus", trust_remote_code=True)
 
 messages = [{"role": "system", "content": "あなたはAIアシスタントです。"}]
 messages.append({"role": "user", "content": "イギリスの首相は誰ですか？"})