Skip to content

Systemcluster/sentencepiece-model

Repository files navigation

sentencepiece-model

Crates.io Docs.rs

SentencePiece model parser generated from the SentencePiece protobuf definition.

use sentencepiece_model::SentencePieceModel;

let model = SentencePieceModel::from_file("tests/t5-spiece.model")?;
assert_eq!(model.pieces.len(), 32000);
assert_eq!(model.trainer()?.unk_id(), 2);

Usage

[dependencies]
sentencepiece-model = "0.1"

sentencepiece-model uses prost-build and protox to generate Rust code from the SentencePiece protobuf definition at build time. protoc is not required.

About

SentencePiece model parser generated from the SentencePiece protobuf definition.

Topics

Resources

License

Stars

Watchers

Forks

Sponsor this project

 

Languages