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
)

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