Update README.md

README.md

@@ -32,7 +32,7 @@ model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
 
 # Load the LoRA adapter
-model = PeftModel.from_pretrained(model, "DDiaa/Unigram-Llama-3.2-3B")
+model = PeftModel.from_pretrained(model, "DDiaa/WM-Removal-Unigram-Llama-3.2-3B")
 
 # Prepare the prompt