From a6032bee35da8e5668db0a9801b02d3f03c77bb0 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Tue, 11 Jul 2023 15:46:41 -0400 Subject: [PATCH] Update CAPTION.md --- doc/CAPTION.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/CAPTION.md b/doc/CAPTION.md index 2f2c9d7..a29ec3f 100644 --- a/doc/CAPTION.md +++ b/doc/CAPTION.md @@ -4,7 +4,7 @@ `python caption_fl.py --data_root input --min_new_tokens 20 --max_new_tokens 30 --num_beams 3 --model "openflamingo/OpenFlamingo-9B-vitl-mpt7b"` -This script uses two example image/caption pairs located in the example folder to prime the system to caption, then captions the images in the input folder. It will save a `.txt` file of the same base filename with the captoin in the same folder. +This script uses two example image/caption pairs located in the `/example` folder to prime the system to caption, then captions the images in the input folder. It will save a `.txt` file of the same base filename with the captoin in the same folder. This script currently requires an AMPERE or newer GPU due to using bfloat16.