candle_transformers/models/nvembed_v2/
mod.rs

1//! NV-Embed-v2
2//!
3//! NV-Embed-v2 is a text embedding model that combines a Mistral decoder with a latent attention mechanism to produce high-quality text embeddings.
4//!
5//! This implementation is based on the [paper](https://arxiv.org/pdf/2405.17428) and [weights](https://huggingface.co/nvidia/NV-Embed-v2)
6//!
7//! # Query-Passage Retrieval Example
8//! ```bash
9//! cargo run --example nvembed_v2 --release
10//! ```
11//!
12//! # Sentence Embedding Example
13//! ```bash
14//! cargo run --example nvembed_v2 --release -- --prompt "Here is a test sentence"
15//! ```
16
17pub mod embedding;
18pub mod model;