Update README.md

README.md

@@ -12,6 +12,7 @@ library_name: peft
 import torch
 from peft import PeftModel
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
+
 # モデルの読み込み
 model = AutoModelForCausalLM.from_pretrained(
     "meta-llama/Llama-2-7b-hf",
@@ -23,10 +24,12 @@ model = AutoModelForCausalLM.from_pretrained(
     ),
     device_map={"":0}
 )
+
 # トークナイザーの読み込み
 tokenizer = AutoTokenizer.from_pretrained(
-    "asaoka/Llama-2-7b-hf-qlora-dolly15k-japanese"
+    "meta-llama/Llama-2-7b-hf"
 )
+
 # LoRAの読み込み
 model = PeftModel.from_pretrained(
     model,
@@ -34,8 +37,10 @@ model = PeftModel.from_pretrained(
     device_map={"":0}
 )
 model.eval()
+
 # プロンプトの準備
 prompt = "### Instruction: 富士山とは？\n\n### Response: "
+
 # 推論の実行
 inputs = tokenizer(prompt, return_tensors="pt").to("cuda:0")
 with torch.no_grad():