dialogue-generation

用XLNet和GPT-2快速训练高质量对话回复模型

这是一个基于PyTorch的对话生成研究项目，利用预训练的XLNet和GPT-2模型来生成自然语言回复。项目旨在解决开放域对话系统中回复质量不高、缺乏连贯性的问题，通过微调大规模预训练语言模型，实现更流畅、更有信息量的对话。核心能力包括：支持XLNet和GPT-2两种主流预训练模型的加载与微调，提供Topical Chat等数据集的预处理脚本，以及训练和推理的完整流程。项目代码结构清晰，适合作为对话生成任务的基线或研究起点。

开源 free 对话助手
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 118
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类对话助手
开发团队bme-chatbots
所属国家
官网地址
GitHubhttps://github.com/bme-chatbots/dialogue-generation
定价模式free
价格说明开源项目，MIT许可证，完全免费，可自行部署使用。
访问状态
是否开源是
开源协议MIT
主要语言Python
技术栈/模型apex,chatbot,dialogue-generation,gpt2,topical-chat,xlnet
GitHub 星标★ 118
30天Star增速
HF 下载量
上线时间2019-04-07 00:00:00
最近更新2024-10-29 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2

使用教程

核心亮点

  • 同时支持XLNet和GPT-2两种预训练模型，方便对比实验
  • 基于PyTorch实现，代码简洁，易于二次开发和扩展
  • 提供Topical Chat数据集处理脚本，降低数据准备门槛

不足之处

  • 文档/社区待观察
  • 项目处于早期阶段，功能相对基础，缺乏高级优化技巧

适用场景

  • 学术研究中对比不同预训练模型在对话生成上的效果
  • 开发者快速搭建对话生成基线系统
  • 教学示例，学习如何微调预训练语言模型

替代项目

ParlAI、Hugging Face Transformers、DialoGPT

项目介绍

dialogue-generation 是对话助手领域的开源项目，由 bme-chatbots 开发，2019 年首次发布。

它收录于对话助手分类，该分类目前共有 907 个项目，GitHub 星标数为 118。

项目已超过三个月没有代码更新，维护节奏明显放缓，最近一次代码更新于 2024-10-29。MIT许可证，完全免费，可自行部署使用。

它主要面向的使用场景是：学术研究中对比不同预训练模型在对话生成上的效果。同类可对比的替代方案包括 ParlAI、Hugging Face Transformers、DialoGPT。

上一篇：Recluse-Bot

下一篇：Streamly

同类项目推荐

open-webui 开源

给本地模型配个漂亮聊天室，全家都能用

User-friendly AI Interface (Supports Ollama, OpenAI API, ...)

★ 152821 2026-08-09
prompts.chat 开源

海量好提示词，抄了就能让 AI 更听话

f.k.a. Awesome ChatGPT Prompts. Share, discover, and collect prompts from the commun···

★ 170999 2026-08-09
elia 开源

终端里用键盘快速聊 AI，多模型切换不打断思路

A snappy, keyboard-centric terminal user interface for interacting with large langua···

★ 2479 2026-08-09
NextChat 开源

一个界面聊遍所有主流AI模型，支持全平台部署，轻快又私密。

✨ Zero-config AI chat assistant. No API key needed — sign up and instantly chat wi···

★ 88802 2026-08-09