diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-10-21 22:44:13 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-21 22:44:13 +0100 |
| commit | 3115fe42e4b203b02219eaf85b749f6710d0de3e (patch) | |
| tree | 2fc86982d579213ef53e6c61ea17f1f8773c6bd4 /candle-examples | |
| parent | 2531b13bf85a69058e8ed1b30c683d19d036df14 (diff) | |
| download | candle-3115fe42e4b203b02219eaf85b749f6710d0de3e.tar.gz candle-3115fe42e4b203b02219eaf85b749f6710d0de3e.tar.bz2 candle-3115fe42e4b203b02219eaf85b749f6710d0de3e.zip | |
Blip attention mask + readme (#1146)
* Add the attention mask to the blip model.
* Add a readme.
Diffstat (limited to 'candle-examples')
| -rw-r--r-- | candle-examples/examples/blip/README.md | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/candle-examples/examples/blip/README.md b/candle-examples/examples/blip/README.md new file mode 100644 index 00000000..e5ceae06 --- /dev/null +++ b/candle-examples/examples/blip/README.md @@ -0,0 +1,19 @@ +# candle-blip + +The +[blip-image-captioning](https://huggingface.co/Salesforce/blip-image-captioning-base) +model can generate captions for an input image. + +## Running on an example + +```bash +cargo run --example blip --release -- --image candle-examples/examples/yolo-v8/assets/bike.jpg +``` + +``` +Running on CPU, to run on GPU, build this example with `--features cuda` +loaded image Tensor[dims 3, 384, 384; f32] +model built +several cyclists are riding down a road with cars behind them% +``` + |