NV-Embed-v2

<p align="center"> <img src="tokie-banner.png" alt="tokie" width="600"> </p>

NV-Embed-v2

Pre-built tokie tokenizer for nvidia/NV-Embed-v2.

Quick Start (Python)

pip install tokie

import tokie

tokenizer = tokie.Tokenizer.from_pretrained("tokiers/NV-Embed-v2")
encoding = tokenizer.encode("Hello, world!")
print(encoding.ids)
print(encoding.attention_mask)

Quick Start (Rust)

[dependencies]
tokie = { version = "0.0.7", features = ["hf"] }

use tokie::Tokenizer;

let tokenizer = Tokenizer::from_pretrained("tokiers/NV-Embed-v2").unwrap();
let encoding = tokenizer.encode("Hello, world!", true);
println!("{:?}", encoding.ids);

Files

• tokenizer.tkz — tokie binary format (~10x smaller, loads in ~5ms)
• tokenizer.json — original HuggingFace tokenizer

About tokie

50x faster tokenization, 10x smaller model files, 100% accurate.

tokie is a drop-in replacement for HuggingFace tokenizers, built in Rust. See GitHub for benchmarks and documentation.

License

MIT OR Apache-2.0 (tokie library). Original model files retain their original license from nvidia/NV-Embed-v2.