README.md

metadata

title: DaiyuLM
emoji: 💬
colorFrom: yellow
colorTo: purple
sdk: gradio
sdk_version: 5.0.1
app_file: app.py
pinned: false
license: apache-2.0
short_description: 林黛玉式“发疯文学”生成模型，基于 ChatGLM2，使用 Prefix Tuning 微调而成。

🌸 DaiyuLM - 林黛玉发疯文学生成模型

DaiyuLM 是一个基于 ChatGLM2-6B 的微调语言模型，致力于生成具有“林黛玉式”风格的情绪化古风回复。它善于表达情绪低落、委屈内敛、怨而不怒的中文对话内容，常被称为“发疯文学”。

本项目使用 Prefix Tuning (P-Tuning v2) 技术在 ChatGLM2 上微调，仅训练少量参数，具备较强的风格适配能力和低资源部署优势。

---

🚀 在线体验

点击右上角「Spaces」运行按钮，输入一句现代中文（如日常对话、争吵等），模型将尝试用“林黛玉风格”进行回应。

📌 示例输入：

• 你是不是不想理我了？
• 你到底还在不在乎我？
• 我多说几句你就不开心了吗？

📜 示例输出：

哥哥若是这般态度，倒不如直接不理我的好，显得我无理取闹了些。

---

🧠 模型细节

• 基座模型：ChatGLM2-6B（支持流式对话）
• 微调方式：Prefix Tuning（128 token embedding）
• 训练语料：142 条“林黛玉风格”对话样本（手工撰写）
• 训练目标：模仿古典情绪表达与委婉怨气风格

---

🧪 使用方式（离线部署）

pip install -r requirements.txt

# 启动 Gradio demo
python app.py

确保你具备：

• ChatGLM2-6B 原始权重（含 tokenizer）
• Prefix Tuning 微调权重 pytorch_model.bin

---

📦 文件结构说明

.
├── app.py                # Gradio Chatbot 前端
├── pytorch_model.bin     # 微调后的 Prefix 参数
├── README.md             # 当前文档
├── requirements.txt      # 环境依赖

---

📄 License

Apache 2.0 License

---

🙏 致谢

• THUDM/ChatGLM2-6B
• Gradio UI

---

👋 如需添加更多人物风格（如宝玉式、凤姐式等）或部署为 API，请联系作者或提 Issue。

📮 联系邮箱：cs.shunzhang@foxmail.com