Update README_zh.md

README_zh.md

@@ -29,8 +29,8 @@ license: other
 悟道·天鹰Aquila系列模型将持续开源更优版本。
 
 - 2023/07/19 ：开源 v0.9
-  - AquilaCode-mutil-01  md5: e202e5b82db773ea369fe843fef1c34c
-  - AquilaCode-mutil-02  md5: 3923b2b020e2af71755b11248076437f
+  - AquilaCode-mutil-01  md5: e6ea49fea7a737ffe41086ec7019cebb
+  - AquilaCode-mutil-02  md5: 4bba98eac44d785358ed5b6d2144a94a
   - AquilaCode-Python-01 md5: e202e5b82db773ea369fe843fef1c34c
   - AquilaCode-Python-02 md5: 3923b2b020e2af71755b11248076437f
     
@@ -49,20 +49,18 @@ Aquila-7B v0.8 在 FlagEval 大模型评测中（ “客观”）相比0.7的版
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-from cyg_conversation import covert_prompt_to_input_ids_with_history
 
-tokenizer = AutoTokenizer.from_pretrained("BAAI/AquilaChat-7B")
-model = AutoModelForCausalLM.from_pretrained("BAAI/AquilaChat-7B")
+model_info = "BAAI/AquilaCode-multi"
+tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
 model.eval()
-model.to("cuda:0")
-vocab = tokenizer.vocab
-print(len(vocab))
+model.to("cuda:3")
 
-text = "请给出10个要到北京旅游的理由。"
+text = "#补全代码\ndef quick_sort(x):"
 
-tokens = covert_prompt_to_input_ids_with_history(text, history=[], tokenizer=tokenizer, max_token=512)
+tokens = tokenizer.encode_plus(text)['input_ids'][:-1]
 
-tokens = torch.tensor(tokens)[None,].to("cuda:0")
+tokens = torch.tensor(tokens)[None,].to("cuda:3")
 
 
 with torch.no_grad():