Update README.md

README.md

@@ -57,17 +57,17 @@ For detailed version change history, see [Change Log](https://huggingface.co/BAA
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-model_info = "BAAI/Aquila-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")
+model.to("cuda:4")
 
-text = "汽车EDR是什么"
+text = "#补全代码\ndef quick_sort(x):"
 
 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():