Commit 6735898e authored by wanglch's avatar wanglch
Browse files

Update README.md

parent e156bfe1
......@@ -112,11 +112,11 @@ pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple
训练需将finetune文件夹中的[finetune_lora.sh](./finetune/finetune_ds.sh)中修改以下参数
```
MODEL="XXXXXX/MiniCPM-Llama3-V-2_5" # or 修改为本地模型地址
DATA="data/self_build/" # 本地自定义训练集json文件
EVAL_DATA="data/self_build/" # 本地自定义验证集json文件
DATA="data/self_build/data/train_data/train_data.json" # 本地自定义训练集json文件
EVAL_DATA="data/self_build/data/eval_data/eval_data.json" # 本地自定义验证集json文件
--output_dir /home/wanglch/projects/saves/MiniCPM-Llama3-V-2_5/lora_train_dtk \
--logging_dir /home/wanglch/projects/saves/MiniCPM-Llama3-V-2_5/lora_train_dtk \
--output_dir /saves/MiniCPM-Llama3-V-2_5/lora_train_dtk \
--logging_dir /saves/MiniCPM-Llama3-V-2_5/lora_train_dtk \
```
### 多卡分布式训练
......@@ -129,7 +129,7 @@ sh finetune_ds.sh
## 推理
执行多种任务时需要对以下参数进行修改
model_path = 'openbmb/MiniCPM-Llama3-V-2_5' # 修改为本地模型路径
model_path = 'XXXXXX/MiniCPM-Llama3-V-2_5' # 修改为本地模型路径
### 单机单卡
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment