Update README.md

README.md

@@ -36,8 +36,8 @@ This model is intended to answer questions about code fragments, to generate cod
 
 ```python
 from transformers import GemmaTokenizer, AutoModelForCausalLM
-tokenizer = GemmaTokenizer.from_pretrained("EpistemeAI/Athene-codegemma-7b-it-alpaca-v1.1")
-model = AutoModelForCausalLM.from_pretrained("EpistemeAI/Athene-codegemma-7b-it-alpaca-v1.1")
+tokenizer = GemmaTokenizer.from_pretrained("EpistemeAI/Athene-codegemma-2-7b-it-alpaca-v1.1")
+model = AutoModelForCausalLM.from_pretrained("EpistemeAI/Athene-codegemma-2-7b-it-alpaca-v1.1")
 input_text = "Write me a Python function to calculate the nth fibonacci number."
 input_ids = tokenizer(input_text, return_tensors="pt")
 outputs = model.generate(**input_ids)
@@ -55,7 +55,7 @@ Let's load the model and apply the chat template to a conversation. In this exam
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import transformers
 import torch
-model_id = "EpistemeAI/Athene-codegemma-2-7b-it-alpaca-v1"
+model_id = "EpistemeAI/Athene-codegemma-2-7b-it-alpaca-v1.1"
 dtype = torch.bfloat16
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(