--- language: - zh library_name: transformers pipeline_tag: text2text-generation --- ```python from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("svjack/gpt-dialogue") model = AutoModelForCausalLM.from_pretrained("svjack/gpt-dialogue") tokenizer.decode( model.generate( tokenizer.encode( "你认为今天的天气怎么样?", return_tensors="pt", add_special_tokens=True ), max_length = 128, num_beams=2, top_p = 0.95, top_k = 50, repetition_penalty = 2.5, length_penalty=1.0, early_stopping=True, )[0], skip_special_tokens = True ) ''' '你 认 为 今 天 的 天 气 怎 么 样 ? 我 认 为 这 会 很 冷 , 但 是 春 天 是 如 此 美 丽 。 您 最 喜 欢 哪 个 季 节 ? 我 真 的 不 知 道 。 也 许 在 山 上 有 一 些 雪 松 石 。 听 起 来 不 错 。 我 们 可 以 去 海 滩 吗 ? 当 然 。 我 们 应 该 尽 快 去 那 里 。 好 的 。 让 我 们 看 看 它 是 否 准 备 好 了 。 现 在 下 雨 了 。 我 们 要 去 哪 里 见 面 ? 我 们' ''' ```