Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 base_model:
-- unsloth/Llama-3.2-1B
+- unsloth/Llama-3.2-1B-Instruct
 library_name: peft
 license: apache-2.0
 language:
@@ -35,8 +35,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load tokenizer and models
 print("Loading models...")
-tokenizer = AutoTokenizer.from_pretrained("unsloth/llama-3.2-1b-instruct-bnb-4bit")
-base_model = AutoModelForCausalLM.from_pretrained("unsloth/llama-3.2-1b-instruct-bnb-4bit").to(device)
+tokenizer = AutoTokenizer.from_pretrained("unsloth/llama-3.2-1b-instruct")
+base_model = AutoModelForCausalLM.from_pretrained("unsloth/llama-3.2-1b-instruct").to(device)
 model = PeftModel.from_pretrained(base_model, "HackWeasel/llama-3.2-1b-QLORA-IMDB").to(device)
 model.eval()
 print("Models loaded!")