Update README.md

README.md

@@ -15,10 +15,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 from peft import PeftModel, PeftConfig
 
 # Load model and tokenizer configurations
-config = PeftConfig.from_pretrained("Vijayendra/llama-3b-merged-4bit")
+config = PeftConfig.from_pretrained("Vijayendra/llama3.0-8B-merged-4bit")
 base_model = AutoModelForCausalLM.from_pretrained("unsloth/llama-3-8b-bnb-4bit")
-model = PeftModel.from_pretrained(base_model, "Vijayendra/llama-3b-merged-4bit")
-tokenizer = AutoTokenizer.from_pretrained("Vijayendra/llama-3b-merged-4bit")
+model = PeftModel.from_pretrained(base_model, "Vijayendra/llama3.0-8B-merged-4bit")
+tokenizer = AutoTokenizer.from_pretrained("Vijayendra/llama3.0-8B-merged-4bit")
 
 # Ensure padding token is set for the tokenizer
 if tokenizer.pad_token is None: