Update README.md

README.md

@@ -49,14 +49,14 @@ Use the code below to get started with the model:
 
 ```python
 import torch
-from transformers import AutoTokenizer, AutoPeftModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 DEV = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-adapter_path = "SungJoo/llama2-7b-sft-detox-DPO"
+adapter_path = "SungJoo/llama2-7b-sft-dpo-detox"
 
 # Load model
-model = AutoPeftModelForCausalLM.from_pretrained(
+model = AutoModelForCausalLM.from_pretrained(
     adapter_path,
     torch_dtype=torch.bfloat16
 ).to(DEV)