[ENH] Restructure the project.

This commit is contained in:
herobrine19
2023-05-22 02:39:04 +08:00
parent f329433cad
commit 46a9cbc21e
22 changed files with 755 additions and 36 deletions

21
scripts/webui.sh Normal file
View File

@ -0,0 +1,21 @@
#!/bin/bash
# 使用huggingface上已经训练好的模型
python webui.py \
--load_8bit True \
--base_model 'minlik/chinese-alpaca-plus-7b-merged' \
--lora_weights 'entity303/lawgpt-lora-7b' \
--prompt_template "law_template" \
--server_name "0.0.0.0" \
--share_gradio Ture \
# 使用自己finetune的lora, 把自己的模型放到对应目录即可
# python webui.py \
# --load_8bit True \
# --base_model 'minlik/chinese-alpaca-plus-7b-merged' \
# --lora_weights './outputs/chinese-alpaca-plus-7b-law-e1' \
# --prompt_template "law_template" \
# --server_name "0.0.0.0" \
# --share_gradio Ture \