Update README.md

README.md

@@ -84,13 +84,13 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Your model's ID on the Hugging Face Hub
-# (In your case: "chachinggg/Mistral-ai-RLHF")
-model_id = "YOUR_MODEL_ID_ON_HUB" 
+
+model_id = "chachinggg/Mistral-ai-RLHF" 
 
 # Load the tokenizer
 # (The padding settings from cell 3 of your notebook
 # should be saved with the model)
-tokenizer = AutoTokenizer.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(Mistral-7B-Instruct-v0.2)
 
 # Load the model (the base model, not the classification head from cell 5)
 # Use the bfloat16 dtype as specified in your notebook