shunxing1234 commited on
Commit
beede7b
1 Parent(s): db57085

Update README_zh.md

Browse files
Files changed (1) hide show
  1. README_zh.md +2 -2
README_zh.md CHANGED
@@ -45,8 +45,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
45
  model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
46
  model.eval()
47
  model.to(device)
48
- text = "请给出10个要到北京旅游的理由。"
49
- tokens = tokenizer.encode_plus(text)['input_ids'][:-1]
50
  tokens = torch.tensor(tokens)[None,].to(device)
51
  stop_tokens = ["###", "[UNK]", "</s>"]
52
  with torch.no_grad():
 
45
  model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
46
  model.eval()
47
  model.to(device)
48
+ text = "明天是个"
49
+ tokens = tokenizer.encode_plus(text)['input_ids']
50
  tokens = torch.tensor(tokens)[None,].to(device)
51
  stop_tokens = ["###", "[UNK]", "</s>"]
52
  with torch.no_grad():