Update README.md

README.md

@@ -30,6 +30,10 @@ tasks:
 - [📊 Benchmark评估/Benchmark Evaluation](#Benchmark)
 - [📜 声明与协议/Terms and Conditions](#Terms)
 
+# 更新
+[2023.12.29] 🎉🎉🎉 我们发布了 **[Baichuan2-13B-Chat](https://huggingface.co/baichuan-inc/Baichuan2-13B-Chat) v2** 版本。其中：
+- 大幅提升了模型的综合能力，特别是数学和逻辑推理、复杂指令跟随能力。
+- 模型处理长度从 4096 提升 至8192 。
 
 # <span id="Introduction">模型介绍/Introduction</span>
 
@@ -59,9 +63,9 @@ In the Baichuan 2 series models, we have utilized the new feature `F.scaled_dot_
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation.utils import GenerationConfig
-tokenizer = AutoTokenizer.from_pretrained("baichuan-inc/Baichuan2-13B-Chat", use_fast=False, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("baichuan-inc/Baichuan2-13B-Chat", device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
-model.generation_config = GenerationConfig.from_pretrained("baichuan-inc/Baichuan2-13B-Chat")
+tokenizer = AutoTokenizer.from_pretrained("baichuan-inc/Baichuan2-13B-Chat", revision="v2.0", use_fast=False, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("baichuan-inc/Baichuan2-13B-Chat", revision="v2.0", device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
+model.generation_config = GenerationConfig.from_pretrained("baichuan-inc/Baichuan2-13B-Chat", revision="v2.0")
 messages = []
 messages.append({"role": "user", "content": "解释一下“温故而知新”"})
 response = model.chat(tokenizer, messages)