diff --git a/scripts/webui.sh b/scripts/webui.sh index 91d05c9..0585006 100644 --- a/scripts/webui.sh +++ b/scripts/webui.sh @@ -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 \ diff --git a/templates/law_template.json b/templates/law_template.json index a90ee9f..ec9ee6b 100644 --- a/templates/law_template.json +++ b/templates/law_template.json @@ -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": "### 回答:" } \ No newline at end of file