From 2d0a167ba3c17f4c1f2561b90ba32de5fb44f52c Mon Sep 17 00:00:00 2001 From: junchen14 Date: Wed, 25 Oct 2023 08:36:31 +0300 Subject: [PATCH] add minigptv2_train md --- MiniGPTv2_Train .md | 4 ++-- README.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/MiniGPTv2_Train .md b/MiniGPTv2_Train .md index 8eadb95..254d680 100644 --- a/MiniGPTv2_Train .md +++ b/MiniGPTv2_Train .md @@ -4,7 +4,7 @@ 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 the 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 following paths: llama_model checkpoint path: "/path/to/llama_checkpoint" @@ -13,7 +13,7 @@ ckpt: "/path/to/pretrained_checkpoint" ckpt save path: "/path/to/save_checkpoint" For ckpt, you may load from our pretrained model checkpoints: -| MiniGPT-v2 (after stage-2) | MiniGPT-v2 (after stage-3) | MiniGPT-v2 (developing model (online demo)) | +| MiniGPT-v2 (after stage-2) | MiniGPT-v2 (after stage-3) | MiniGPT-v2 (online developing demo) | |------------------------------|------------------------------|------------------------------| | [Download](https://drive.google.com/file/d/1Vi_E7ZtZXRAQcyz4f8E6LtLh2UXABCmu/view?usp=sharing) |[Download](https://drive.google.com/file/d/1jAbxUiyl04SFJMN4sF1vvUU69Etuz4qa/view?usp=sharing) | [Download](https://drive.google.com/file/d/1aVbfW7nkCSYx99_vCRyP1sOlQiWVSnAl/view?usp=sharing) | diff --git a/README.md b/README.md index d75463b..8459936 100644 --- a/README.md +++ b/README.md @@ -93,9 +93,9 @@ Then, set the variable *llama_model* in the model config file to the LLM weight Download the pretrained model checkpoints -| MiniGPT-v2 (developing model (online demo)) | MiniGPT-v2 (after stage-2) | MiniGPT-v2 (after stage-3) +| MiniGPT-v2 (after stage-2) | MiniGPT-v2 (after stage-3) | MiniGPT-v2 (online developing demo)| |------------------------------|------------------------------|------------------------------| -| [Download](https://drive.google.com/file/d/1aVbfW7nkCSYx99_vCRyP1sOlQiWVSnAl/view?usp=sharing) |[Download](https://drive.google.com/file/d/1Vi_E7ZtZXRAQcyz4f8E6LtLh2UXABCmu/view?usp=sharing) |[Download](https://drive.google.com/file/d/1jAbxUiyl04SFJMN4sF1vvUU69Etuz4qa/view?usp=sharing) | +| [Download](https://drive.google.com/file/d/1Vi_E7ZtZXRAQcyz4f8E6LtLh2UXABCmu/view?usp=sharing) |[Download](https://drive.google.com/file/d/1jAbxUiyl04SFJMN4sF1vvUU69Etuz4qa/view?usp=sharing) | [Download](https://drive.google.com/file/d/1aVbfW7nkCSYx99_vCRyP1sOlQiWVSnAl/view?usp=sharing) | For **MiniGPT-v2**, set the path to the pretrained checkpoint in the evaluation config file