From f58149df4fbc81aaacde9c28e3d50afc1f7b3676 Mon Sep 17 00:00:00 2001 From: junchen14 Date: Wed, 25 Oct 2023 08:32:56 +0300 Subject: [PATCH] add minigptv2_train md --- MiniGPTv2_Train .md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/MiniGPTv2_Train .md b/MiniGPTv2_Train .md index db30220..8eadb95 100644 --- a/MiniGPTv2_Train .md +++ b/MiniGPTv2_Train .md @@ -4,9 +4,12 @@ You firstly need to prepare the dataset. you can follow this step to prepare the dataset. our [dataset preparation](dataset/README_MINIGPTv2_FINETUNE.md). -in train_configs/minigptv2_finetune.yaml, you need to set up the paths +In the train_configs/minigptv2_finetune.yaml, you need to set up the paths: + llama_model checkpoint path: "/path/to/llama_checkpoint" + ckpt: "/path/to/pretrained_checkpoint" + ckpt save path: "/path/to/save_checkpoint" For ckpt, you may load from our pretrained model checkpoints: