支持多种情绪风格的对话生成,涵盖数据自动生成、格式转换、模型微调、推理部署和 Web 交互等完整流程。适用于情感陪伴、虚拟助手等场景。
- 设计多情绪风格对话模板体系,构建高质量训练数据集,增强模型情感理解与表达能力
- 完成数据格式标准化处理,确保与LLaMA Factory框架的完美兼容,支持灵活的单轮/多轮对话训练
- 严格按照LLaMA Factory官方最佳实践完成模型微调,建立规范化的训练流程
- 通过vLLM本地部署优化推理性能,结合Gradio开发直观的Web界面,支持实时参数调优
# 进入 EmotionalDialogueModel 目录
cd EmotionalDialogueModel
# 生成多风格对话数据
python generate_data.py
# 转换为 LlamaFactory 格式
python data_convert.py
# 部署 vLLM 并启动 Web UI
python webui.py
点击空白处退出提示
评论