diff --git a/README.md b/README.md
index a0e6757..aaf2742 100644
--- a/README.md
+++ b/README.md
@@ -24,14 +24,14 @@ LaWGPT 是一系列基于中文法律知识的开源大语言模型。
## 更新
-- 🌟 2023/05/13:公开发布
+- 🌟 2023/05/13:公开发布
- - **legal-base-7b**:法律基座模型,使用 50w 中文裁判文书数据二次预训练
+ - **Legal-Base-7B**:法律基座模型,使用 50w 中文裁判文书数据二次预训练
- - **lawgpt-7b-beta1.0**:法律对话模型,构造 30w 高质量法律问答数据集指令精调
+ - **LaWGPT-7B-beta1.0**:法律对话模型,构造 30w 高质量法律问答数据集指令精调
-- 🌟 2023/04/12:内部测试
- - **lawgpt-7b-alpha**:在 Chinese-LLaMA-7B 的基础上直接构造 30w 法律问答数据集指令精调
+- 🌟 2023/04/12:内部测试
+ - **LaWGPT-7B-alpha**:在 Chinese-LLaMA-7B 的基础上直接构造 30w 法律问答数据集指令精调
## 快速开始
@@ -44,8 +44,12 @@ LaWGPT 是一系列基于中文法律知识的开源大语言模型。
pip install -r requirements.txt
```
-2. 合并模型权重
+2. 合并模型权重(可选)
+ 如果您想使用 LaWGPT-7B-alpha 模型,可跳过改步,直接进入步骤3.
+
+ 如果您想使用 LaWGPT-7B-beta1.0 模型:
+
由于 [LLaMA](https://github.com/facebookresearch/llama) 和 [Chinese-LLaMA](https://github.com/ymcui/Chinese-LLaMA-Alpaca) 均未开源模型权重。根据相应开源许可,**本项目只能发布 LoRA 权重**,无法发布完整的模型权重,请各位谅解。
本项目给出[合并方式](https://github.com/pengxiao-song/LaWGPT/wiki/%E6%A8%A1%E5%9E%8B%E5%90%88%E5%B9%B6),请各位获取原版权重后自行重构模型。
@@ -61,14 +65,7 @@ LaWGPT 是一系列基于中文法律知识的开源大语言模型。
sh src/scripts/generate.sh
```
- 接入服务链接:
-
- ```
- Running on local URL: http://0.0.0.0:7862
- Running on public URL: https://06e989c08fe171f47c.gradio.live
- ```
-
- 呈现效果:
+ 接入服务,呈现效果:
@@ -101,9 +98,11 @@ LaWGPT
## 数据构建
-本项目汇总了互联网上的中文法律数据源
+本项目汇总互联网上的中文法律数据源
-根据 [Stanford_alpaca](https://github.com/tatsu-lab/stanford_alpaca#data-generation-process) 和 [self-instruct](https://github.com/yizhongw/self-instruct) 方式数据生成
+1. 初步生成数据:根据 [Stanford_alpaca](https://github.com/tatsu-lab/stanford_alpaca#data-generation-process) 和 [self-instruct](https://github.com/yizhongw/self-instruct) 方式生成对话问答数据
+2. 基于知识生成数据:通过 Knowledge-based Self-Instruct 方式基于中文法律结构化知识生成数据。
+3. 引入 ChatGPT 清洗数据,并辅助构造高质量数据集。
## 模型训练
@@ -114,9 +113,12 @@ LaWGPT
### 二次训练流程
+1. 参考 `data/example_instruction_train.json` 构造指令微调数据集
+2. 运行 `src/scripts/finetune.sh`
+
### 指令精调步骤
-1. 参考 `data/example_instruction.json` 构造指令微调数据集
+1. 参考 `data/example_instruction_tune.json` 构造指令微调数据集
2. 运行 `src/scripts/finetune.sh`
### 计算资源
@@ -127,6 +129,24 @@ LaWGPT
### 输出示例
+问题:欠了信用卡的钱还不上要坐牢吗?
+
+
+
+问题:民间借贷受国家保护的合法利息是多少?
+
+
+
+问题:你能否写一段抢劫罪罪名的案情描述?
+
+
+
+