caixiaoshun commited on
Commit
7adf610
·
verified ·
1 Parent(s): 523619a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +78 -1
README.md CHANGED
@@ -11,4 +11,81 @@ license: apache-2.0
11
  short_description: 林黛玉式“发疯文学”生成模型,基于 ChatGLM2,使用 Prefix Tuning 微调而成。
12
  ---
13
 
14
- An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+