Update README.md

README.md

@@ -34,7 +34,7 @@ tasks:
 [2023.12.29] 🎉🎉🎉 我们发布了 **[Baichuan2-13B-Chat](https://huggingface.co/baichuan-inc/Baichuan2-13B-Chat) v2** 版本。其中：
 - 大幅提升了模型的综合能力，特别是数学和逻辑推理、复杂指令跟随能力。
 - 模型处理长度从 4096 提升 至8192 。
-- 使用方法参考[快速开始](#Start)
+- 使用时需指定revision=v2.0，详细方法参考[快速开始](#Start)
 
 # <span id="Introduction">模型介绍/Introduction</span>
 
@@ -64,8 +64,15 @@ 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", 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)
+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": "解释一下“温故而知新”"})