Spaces:
Runtime error
Runtime error
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,4 +11,81 @@ license: apache-2.0
|
|
| 11 |
short_description: 林黛玉式“发疯文学”生成模型,基于 ChatGLM2,使用 Prefix Tuning 微调而成。
|
| 12 |
---
|
| 13 |
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
short_description: 林黛玉式“发疯文学”生成模型,基于 ChatGLM2,使用 Prefix Tuning 微调而成。
|
| 12 |
---
|
| 13 |
|
| 14 |
+
# 🌸 DaiyuLM - 林黛玉发疯文学生成模型
|
| 15 |
+
|
| 16 |
+
**DaiyuLM** 是一个基于 [ChatGLM2-6B](https://huggingface.co/THUDM/chatglm2-6b) 的微调语言模型,致力于生成具有“林黛玉式”风格的情绪化古风回复。它善于表达情绪低落、委屈内敛、怨而不怒的中文对话内容,常被称为“发疯文学”。
|
| 17 |
+
|
| 18 |
+
本项目使用 [Prefix Tuning (P-Tuning v2)](https://arxiv.org/abs/2110.07602) 技术在 ChatGLM2 上微调,仅训练少量参数,具备较强的风格适配能力和低资源部署优势。
|
| 19 |
+
|
| 20 |
+
---
|
| 21 |
+
|
| 22 |
+
## 🚀 在线体验
|
| 23 |
+
|
| 24 |
+
点击右上角「Spaces」运行按钮,输入一句现代中文(如日常对话、争吵等),模型将尝试用“林黛玉风格”进行回应。
|
| 25 |
+
|
| 26 |
+
📌 示例输入:
|
| 27 |
+
|
| 28 |
+
- `你是不是不想理我了?`
|
| 29 |
+
- `你到底还在不在乎我?`
|
| 30 |
+
- `我多说几句你就不开心了吗?`
|
| 31 |
+
|
| 32 |
+
📜 示例输出:
|
| 33 |
+
|
| 34 |
+
> 哥哥若是这般态度,倒不如直接不理我的好,显得我无理取闹了些。
|
| 35 |
+
|
| 36 |
+
---
|
| 37 |
+
|
| 38 |
+
## 🧠 模型细节
|
| 39 |
+
|
| 40 |
+
- 基座模型:`ChatGLM2-6B`(支持流式对话)
|
| 41 |
+
- 微调方式:Prefix Tuning(128 token embedding)
|
| 42 |
+
- 训练语料:142 条“林黛玉风格”对话样本(手工撰写)
|
| 43 |
+
- 训练目标:模仿古典情绪表达与委婉怨气风格
|
| 44 |
+
|
| 45 |
+
---
|
| 46 |
+
|
| 47 |
+
## 🧪 使用方式(离线部署)
|
| 48 |
+
|
| 49 |
+
```bash
|
| 50 |
+
pip install -r requirements.txt
|
| 51 |
+
|
| 52 |
+
# 启动 Gradio demo
|
| 53 |
+
python app.py
|
| 54 |
+
````
|
| 55 |
+
|
| 56 |
+
确保你具备:
|
| 57 |
+
|
| 58 |
+
* ChatGLM2-6B 原始权重(含 tokenizer)
|
| 59 |
+
* Prefix Tuning 微调权重 `pytorch_model.bin`
|
| 60 |
+
|
| 61 |
+
---
|
| 62 |
+
|
| 63 |
+
## 📦 文件结构说明
|
| 64 |
+
|
| 65 |
+
```
|
| 66 |
+
.
|
| 67 |
+
├── app.py # Gradio Chatbot 前端
|
| 68 |
+
├── pytorch_model.bin # 微调后的 Prefix 参数
|
| 69 |
+
├── README.md # 当前文档
|
| 70 |
+
├── requirements.txt # 环境依赖
|
| 71 |
+
```
|
| 72 |
+
|
| 73 |
+
---
|
| 74 |
+
|
| 75 |
+
## 📄 License
|
| 76 |
+
|
| 77 |
+
Apache 2.0 License
|
| 78 |
+
|
| 79 |
+
---
|
| 80 |
+
|
| 81 |
+
## 🙏 致谢
|
| 82 |
+
|
| 83 |
+
* [THUDM/ChatGLM2-6B](https://huggingface.co/THUDM/chatglm2-6b)
|
| 84 |
+
* [Gradio UI](https://gradio.app/)
|
| 85 |
+
|
| 86 |
+
---
|
| 87 |
+
|
| 88 |
+
👋 如需添加更多人物风格(如宝玉式、凤姐式等)或部署为 API,请联系作者或提 Issue。
|
| 89 |
+
|
| 90 |
+
📮 联系邮箱:cs.shunzhang@foxmail.com
|
| 91 |
+
|