Update README.md

README.md

@@ -70,7 +70,7 @@ pip install transformers vllm torch accelerate
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "xl-zhao/PromptCoT-Mamba-7B"
+model_name = "xl-zhao/PromptCoT-Mamba-Math-7B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda")
 
@@ -99,7 +99,7 @@ print(generated_solution)
 ```python
 from vllm import LLM, SamplingParams
 
-model_name = "xl-zhao/PromptCoT-Mamba-7B"
+model_name = "xl-zhao/PromptCoT-Mamba-Math-7B"
 llm = LLM(model=model_name, tensor_parallel_size=1)
 
 problem_statement = (