引言
上篇文章我们看到了如何从零训练一个聊天机器人,本文在此基础上介绍各种生成策略的原理和实现。最后通过Gradio构建一个聊天机器人应用。
体验地址提前放出: https://huggingface.co/spaces/greyfoss/gpt2-chatbot
效果如上图所示。
定义生成框架
def generate(
model,
tokenizer,
prompt,
max_length=