Update README.md

README.md

@@ -46,12 +46,20 @@ from peft import PeftModel
 base_model = "google/gemma-3-1b-it"
 
 tokenizer = AutoTokenizer.from_pretrained(base_model)
+tokenizer.add_special_tokens({
+    "additional_special_tokens": ["<start_of_turn>", "<end_of_turn>"]
+})
 
 model = AutoModelForCausalLM.from_pretrained(
     base_model,
     torch_dtype=torch.bfloat16,
     device_map="auto"
 )
+model.config.pad_token_id = tokenizer.pad_token_id
+model.config.bos_token_id = tokenizer.bos_token_id
+model.config.eos_token_id = tokenizer.eos_token_id
+
+model.resize_token_embeddings(len(tokenizer))
 
 model = PeftModel.from_pretrained(model, f"/lora_adapter")