Commit
•
347d85c
1
Parent(s):
beede7b
Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
|
|
| 39 |
model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
|
| 40 |
model.eval()
|
| 41 |
model.to(device)
|
| 42 |
-
text = "
|
| 43 |
tokens = tokenizer.encode_plus(text)['input_ids']
|
| 44 |
tokens = torch.tensor(tokens)[None,].to(device)
|
| 45 |
stop_tokens = ["###", "[UNK]", "</s>"]
|
|
|
|
| 39 |
model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
|
| 40 |
model.eval()
|
| 41 |
model.to(device)
|
| 42 |
+
text = "杭州亚运会的亮点和期待 2023年9月23日至10月8日，杭州将举办第19届亚洲运动会"
|
| 43 |
tokens = tokenizer.encode_plus(text)['input_ids']
|
| 44 |
tokens = torch.tensor(tokens)[None,].to(device)
|
| 45 |
stop_tokens = ["###", "[UNK]", "</s>"]
|