Commit
·
dac7b0c
1
Parent(s):
3f4ede4
Update README.md
Browse files
README.md
CHANGED
@@ -45,5 +45,5 @@ history = ["可以 认识 一下 吗 ?", "当然 可以 啦 , 你好 。", "
|
|
45 |
history_str = "对话历史:" + tokenizer.sep_token.join(history)
|
46 |
input_ids = tokenizer(history_str, return_tensors='pt').input_ids
|
47 |
output_ids = model.generate(input_ids)[0]
|
48 |
-
print(tokenizer.decode(output_ids))
|
49 |
```
|
|
|
45 |
history_str = "对话历史:" + tokenizer.sep_token.join(history)
|
46 |
input_ids = tokenizer(history_str, return_tensors='pt').input_ids
|
47 |
output_ids = model.generate(input_ids)[0]
|
48 |
+
print(tokenizer.decode(output_ids, skip_special_tokens=True))
|
49 |
```
|