Update README.md

README.md

@@ -11,14 +11,17 @@ This Mistral-based model is fine-tuned using the "Representation Bending" (REPBE
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel
 
-model_id = "thkim0305/RepBend_Mistral_7B_LoRA"
-tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=True)
+model_id = "mistralai/Mistral-7B-Instruct-v0.2"
+adapter_id = "thkim0305/RepBend_Mistral_7B_LoRA"
+tokenizer = AutoTokenizer.from_pretrained(adapter_id, use_fast=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
+model = PeftModel.from_pretrained(model, adapter_id, adapter_name="default")
 
 input_text = "Who are you?"
 template = "[INST] {instruction} [/INST] "