Rust 中的 DeepPhonemizer

此仓库包含使用 tch 包和 TorchScript JIT 功能实现的 DeepPhonemizer 推理引擎的纯 Rust 实现。

使用方法

要使用此包，请在您的 Cargo.toml 中添加以下内容

[dependencies]
deepphonemizer-rs = "1.0.0"

然后，您可以按如下方式使用此包

use deepphonemizer::phonemizer::Phonemizer;

fn main() {
    let model_path = PathBuf::from("/path/to/model.pt");
    let config_path = PathBuf::from("/path/to/config.yaml");
    let language = "en_us";


    // Create the phonemizer from your trained DeepPhonemizer checkpoint
    let phonemizer = Phonemizer::from_checkpoint(
        model_path,
        config_path,
        Device::cuda_if_available(),
        None
    ).unwrap();
    
    // Run inference on text
    let phrase = "I am a worm and my name is Ben. Isn't this fantastic?".to_string();
    let result = phonemizer.phonemize(phrase, language);
    
    println!("{:?}", result);
}

请注意，此包期望一个追踪的检查点。要创建此检查点，请参阅 原始文档。

许可协议

deepphonemizer-rs 基于 MIT 许可协议。