From 525970b7ea6504fbc530c2971846da0ab73bb8df Mon Sep 17 00:00:00 2001 From: XiaoqianShen <64844805+xiaoqian-shen@users.noreply.github.com> Date: Thu, 20 Apr 2023 19:36:57 +0300 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index b712e6d..e092782 100644 --- a/README.md +++ b/README.md @@ -84,6 +84,8 @@ Try out our demo [demo.py](demo.py) on your local machine by running python demo.py --cfg-path eval_configs/minigpt4_eval.yaml --gpu-id 0 ``` +Thanks [@camenduru](https://github.com/camenduru), you can also run our code on [Colab](https://colab.research.google.com/github/camenduru/MiniGPT-4-colab/blob/main/minigpt4_colab.ipynb) + Here, we load Vicuna as 8 bit by default to save some GPU memory usage. Besides, the default beam search width is 1. Under this setting, the demo cost about 23G GPU memory.