File size: 985 Bytes
e657112
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
926a92b
e657112
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
---
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
)

'''
'今 天 天 气 不 错 。 真 好 , 不 是 吗 ?
是 的 , 现 在 的 天 气 很 好 我 们 去 公 园 里 的 那 个 风 景 区 看 看 吧 我 觉 得 那 个 风 景 很 不 错 。
那 是 什 么 样 子 的 公 园 ? 那 是 一 个 以 自 然 风 光 而 闻 名 的 公 园 。
我 想 去 那 里 爬 山 。 我 们 可 以 在 那 里 吃 点 东 西 。
好 啊 我 们 可 以 在 公 园 的 停 车 楼 停 一 下 。 我 想 我'
'''

```