Update README.md

README.md

@@ -47,7 +47,7 @@ We compare AceMath to leading proprietary and open-access math models in above T
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "nvidia/AceMath-7B-Instruct"
+model_name = "nvidia/AceMath-1.5B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")