Update README.md

README.md

@@ -52,7 +52,7 @@ print(len(vocab))
 
 text = "请给出10个要到北京旅游的理由。"
 
-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")