shunxing1234
commited on
Commit
•
beede7b
1
Parent(s):
db57085
Update README_zh.md
Browse files- 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 = "
|
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():
|
|
|
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():
|