Update README.md

README.md

@@ -79,8 +79,8 @@ To use LLMLit, install the required libraries and load the model as follows:
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Load the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("llmlit/Llama-3.1-8B-Instruct")
-tokenizer = AutoTokenizer.from_pretrained("llmlit/Llama-3.1-8B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("llmlit/LLMLit-0.2-8B-Instruct")
+tokenizer = AutoTokenizer.from_pretrained("llmlit/LLMLit-0.2-8B-Instruct")
 
 # Generate text
 inputs = tokenizer("Your prompt here", return_tensors="pt")