From 84f86256451ad8d8daded379ea272a15d2f9e24c Mon Sep 17 00:00:00 2001 From: thinksoso Date: Sun, 30 Apr 2023 20:14:29 +0800 Subject: [PATCH] update --- README.md | 2 ++ scripts/sft_medchat.sh | 2 +- wandb/settings | 2 ++ 3 files changed, 5 insertions(+), 1 deletion(-) create mode 100644 wandb/settings diff --git a/README.md b/README.md index f8e301b..8b245e9 100644 --- a/README.md +++ b/README.md @@ -10,6 +10,8 @@ 基于相同的数据,我们还训练了医疗版本的LLaMA模型: [华驼](https://github.com/SCIR-HI/Huatuo-Llama-Med-Chinese) ## A Quick Start +Updata: 由于chatglm更新很快,推荐使用chatglm官方的微调方案:https://github.com/THUDM/ChatGLM-6B/blob/main/ptuning/README.md + 首先安装依赖包,python环境建议3.9+ ``` diff --git a/scripts/sft_medchat.sh b/scripts/sft_medchat.sh index 4e1cdfa..19adbe2 100644 --- a/scripts/sft_medchat.sh +++ b/scripts/sft_medchat.sh @@ -2,7 +2,7 @@ wandb online exp_tag="chatglm_tuning" python run_clm.py \ - --model_name_or_path MODEL_PATH\ + --model_name_or_path MODEL_PATH \ --per_device_train_batch_size 8 \ --per_device_eval_batch_size 8 \ --train_file ./data/train.txt \ diff --git a/wandb/settings b/wandb/settings new file mode 100644 index 0000000..8dae664 --- /dev/null +++ b/wandb/settings @@ -0,0 +1,2 @@ +[default] +