Skip to content

Commit

Permalink
[ENH] Update lora model.
Browse files Browse the repository at this point in the history
  • Loading branch information
herobrine19 committed May 29, 2023
1 parent c8d1b92 commit 568cafd
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions scripts/webui.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,9 @@

# 使用huggingface上已经训练好的模型
python webui.py \
--load_8bit True \
--base_model 'minlik/chinese-llama-7b-merged' \
--lora_weights 'entity303/lawgpt-lora-7b' \
--load_8bit False \
--base_model 'minlik/chinese-alpaca-plus-7b-merged' \
--lora_weights 'entity303/lawgpt-lora-7b-v2' \
--prompt_template "law_template" \
--server_name "0.0.0.0" \
--share_gradio True \
Expand Down
4 changes: 2 additions & 2 deletions templates/law_template.json
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
{
"description": "Template used by Law Instruction Tuning",
"prompt_input": "下面是一个问题,运用法学知识来正确回答提问.\n### 问题:\n{instruction}\n### 回答:\n",
"prompt_no_input": "下面是一个问题,运用法学知识来正确回答提问.\n### 问题:\n{instruction}\n### 回答:\n",
"prompt_input": "你是中国顶尖智能法律顾问 LaWGPT,具备强大的中文法律基础语义理解能力,能够出色地理解和执行与法律问题和指令。你只能回答与中国法律领域相关的问题,其余领域的问题请礼貌地拒绝回答。接下来,请依据中国法律来回答下面这个问题。\n### 问题:\n{instruction}\n### 回答:\n",
"prompt_no_input": "你是中国顶尖智能法律顾问 LaWGPT,具备强大的中文法律基础语义理解能力,能够出色地理解和执行与法律问题和指令。你只能回答与中国法律领域相关的问题,其余领域的问题请礼貌地拒绝回答。接下来,请依据中国法律来回答下面这个问题。\n### 问题:\n{instruction}\n### 回答:\n",
"response_split": "### 回答:"
}

0 comments on commit 568cafd

Please sign in to comment.