Update README.md

add stage 1 7b link
Vision-CAIR · Apr 27, 2023 · c6acff2 · c6acff2
1 parent a0b7c12
commit c6acff2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -121,7 +121,7 @@ torchrun --nproc-per-node NUM_GPU train.py --cfg-path train_configs/minigpt4_sta
 ```
 
 A MiniGPT-4 checkpoint with only stage one training can be downloaded 
-[here](https://drive.google.com/file/d/1u9FRRBB3VovP1HxCAlpD9Lw4t4P6-Yq8/view?usp=share_link).
+[here (13B)](https://drive.google.com/file/d/1u9FRRBB3VovP1HxCAlpD9Lw4t4P6-Yq8/view?usp=share_link) or [here (7B)](https://drive.google.com/file/d/1HihQtCEXUyBM1i9DQbaK934wW3TZi-h5/view?usp=share_link).
 Compared to the model after stage two, this checkpoint generate incomplete and repeated sentences frequently.