Update README_zh.md

README_zh.md

@@ -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-py"
+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:4")
 
-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:4")
 
 
 with torch.no_grad():