Update README.md

README.md

@@ -15,7 +15,7 @@ language:
 <em>[Paper][Code][🤗] (would be released soon)</em>
 </p>
 
-Infinity-Instruct-7M-0729-Llama3-8B is an opensource supervised instruction tuning model without reinforcement learning from human feedback (RLHF). This model is just finetuned on [Infinity-Instruct-7M and Infinity-Instruct-0729](https://huggingface.co/datasets/BAAI/Infinity-Instruct) and showing favorable results on AlpacaEval 2.0 compared to GPT4.
+Infinity-Instruct-7M-0729-Llama3.1-8B is an opensource supervised instruction tuning model without reinforcement learning from human feedback (RLHF). This model is just finetuned on [Infinity-Instruct-7M and Infinity-Instruct-0729](https://huggingface.co/datasets/BAAI/Infinity-Instruct) and showing favorable results on AlpacaEval 2.0 compared to GPT4.
 
 ## **News**
 
@@ -39,7 +39,7 @@ Infinity-Instruct-7M-0729-Llama3-8B is an opensource supervised instruction tuni
 <img src="fig/trainingflow.png">
 </p>
 
-Infinity-Instruct-7M-0729-Llama3-8B is tuned on Million-level instruction dataset [Infinity-Instruct](https://huggingface.co/datasets/BAAI/Infinity-Instruct). First, we apply the foundational dataset Infinity-Instruct-7M to improve the foundational ability (math & code) of Llama3-8B, and get the foundational instruct model Infinity-Instruct-7M-Llama3-8B. Then we finetune the Infinity-Instruct-7M-Llama3-8B to get the stronger chat model Infinity-Instruct-7M-0729-Llama3-8B. Here is the training hyperparamers. 
+Infinity-Instruct-7M-0729-Llama3.1-8B is tuned on Million-level instruction dataset [Infinity-Instruct](https://huggingface.co/datasets/BAAI/Infinity-Instruct). First, we apply the foundational dataset Infinity-Instruct-7M to improve the foundational ability (math & code) of Llama3-8B, and get the foundational instruct model Infinity-Instruct-7M-Llama3.1-8B. Then we finetune the Infinity-Instruct-7M-Llama3.1-8B to get the stronger chat model Infinity-Instruct-7M-0729-Llama3.1-8B. Here is the training hyperparamers. 
 
 ```bash
 epoch: 3
@@ -71,7 +71,7 @@ Thanks to [FlagScale](https://github.com/FlagOpen/FlagScale), we could concatena
 
 ## **How to use**
 
-Infinity-Instruct-7M-0729-Llama3-8B adopt the same chat template of [Llama3-8B-instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct):
+Infinity-Instruct-7M-0729-Llama3.1-8B adopt the same chat template of [Llama3-8B-instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct):
 
 ```bash
 <|begin_of_text|><|start_header_id|>user<|end_header_id|>
@@ -90,11 +90,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, LogitsProcessorLis
 import torch
 device = "cuda" # the device to load the model onto
 
-model = AutoModelForCausalLM.from_pretrained("BAAI/Infinity-Instruct-7M-0729-Llama3-8B",
+model = AutoModelForCausalLM.from_pretrained("BAAI/Infinity-Instruct-7M-0729-Llama3_1-8B",
     torch_dtype=torch.bfloat16,
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("BAAI/Infinity-Instruct-7M-0729-Llama3-8B")
+tokenizer = AutoTokenizer.from_pretrained("BAAI/Infinity-Instruct-7M-0729-Llama3_1-8B")
 
 prompt = "Give me a short introduction to large language model."
 messages = [