update readme

This commit is contained in:
junchen14 2023-10-13 09:40:14 +03:00
parent 91e6942f70
commit 3c13d1d4b4

View File

@ -38,15 +38,15 @@ More examples can be found in the [project page](https://minigpt-4.github.io).
## Introduction
<!-- ## Introduction
- MiniGPT-4 aligns a frozen visual encoder from BLIP-2 with a frozen LLM, Vicuna, using just one projection layer.
- We train MiniGPT-4 with two stages. The first traditional pretraining stage is trained using roughly 5 million aligned image-text pairs in 10 hours using 4 A100s. After the first stage, Vicuna is able to understand the image. But the generation ability of Vicuna is heavily impacted.
- To address this issue and improve usability, we propose a novel way to create high-quality image-text pairs by the model itself and ChatGPT together. Based on this, we then create a small (3500 pairs in total) yet high-quality dataset.
- The second finetuning stage is trained on this dataset in a conversation template to significantly improve its generation reliability and overall usability. To our surprise, this stage is computationally efficient and takes only around 7 minutes with a single A100.
- MiniGPT-4 yields many emerging vision-language capabilities similar to those demonstrated in GPT-4.
- MiniGPT-4 yields many emerging vision-language capabilities similar to those demonstrated in GPT-4. -->
![overview](figs/overview.png)
<!-- ![overview](figs/overview.png) -->
## Getting Started