mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00

Files

Santiago Medina ace282e5c2 Add flag to run Moondream in f16 precision (#2015 )

* moondream implementation

* add moondream example

* change config default activation

* Add assets and integrate phi mixformer with example

* Make use of kv cache and fix seq_len bug; Clean up example code

* Add README link to example

* Remove pos_embed scaling; Remove assets; Add to README; Expand VisionConfig

* Delete image

* Use apply instead of forward

* Use latest release special token; Fix token/s accuracy; Use GeluPytorchTanh in VisionConfig v2

* Add flag to use f16

* Avoid breaking the quantized version on cuda.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>

2024-04-05 07:03:33 +02:00

main.rs

Add flag to run Moondream in f16 precision (#2015 )

2024-04-05 07:03:33 +02:00

README.md

Add Moondream transformer implementation and example (#1970 )

2024-03-31 08:54:56 +02:00

README.md

candle-moondream

Moondream is a computer-vision model can answer real-world questions about images. It's tiny by today's models, with only 1.6B parameters. That enables it to run on a variety of devices, including mobile phones and edge devices.

Running some examples

First download an example image

$ wget https://raw.githubusercontent.com/vikhyat/moondream/main/assets/demo-1.jpg

Now you can run Moondream from the candle-examples crate:

$ cargo run --example moondream --release -- --prompt "What is the girl eating?" --image "./demo-1.jpg"

avavx: false, neon: true, simd128: false, f16c: false
temp: 0.00 repeat-penalty: 1.00 repeat-last-n: 64
retrieved the files in 3.395583ms
Running on CPU, to run on GPU(metal), build this example with `--features metal`
loaded the model in 5.485493792s
loaded and encoded the image Tensor[dims 3, 378, 378; f32] in 4.801396417s
starting the inference loop
 The girl is eating a hamburger.<
9 tokens generated (0.68 token/s)