|
|
--- |
|
|
language: |
|
|
- zh |
|
|
library_name: transformers |
|
|
pipeline_tag: text2text-generation |
|
|
--- |
|
|
|
|
|
```python |
|
|
|
|
|
from transformers import AutoTokenizer, AutoModelForCausalLM |
|
|
|
|
|
tokenizer = AutoTokenizer.from_pretrained("svjack/gpt-daliy-dialogue") |
|
|
|
|
|
model = AutoModelForCausalLM.from_pretrained("svjack/gpt-daliy-dialogue") |
|
|
|
|
|
tokenizer.decode( |
|
|
model.generate( |
|
|
tokenizer.encode( |
|
|
"今天天气不错。", return_tensors="pt", add_special_tokens=True |
|
|
), max_length = 128)[0], |
|
|
skip_special_tokens = True |
|
|
) |
|
|
|
|
|
''' |
|
|
'今 天 天 气 不 错 。 真 好 ， 不 是 吗 ？ |
|
|
是 的 ， 现 在 的 天 气 很 好 我 们 去 公 园 里 的 那 个 风 景 区 看 看 吧 我 觉 得 那 个 风 景 很 不 错 。 |
|
|
那 是 什 么 样 子 的 公 园 ？ 那 是 一 个 以 自 然 风 光 而 闻 名 的 公 园 。 |
|
|
我 想 去 那 里 爬 山 。 我 们 可 以 在 那 里 吃 点 东 西 。 |
|
|
好 啊 我 们 可 以 在 公 园 的 停 车 楼 停 一 下 。 我 想 我' |
|
|
''' |
|
|
|
|
|
``` |
