Yabin Li
2023-12-26 b8825902d93d5017e44828316062dc8306b7ddcd
runtime/onnxruntime/src/paraformer.cpp
@@ -193,8 +193,7 @@
        lm_ = std::shared_ptr<fst::Fst<fst::StdArc>>(
            fst::Fst<fst::StdArc>::Read(lm_file));
        if (lm_){
            if (vocab) { delete vocab; }
            vocab = new Vocab(lm_cfg_file.c_str(), lex_file.c_str());
            lm_vocab = new Vocab(lm_cfg_file.c_str(), lex_file.c_str());
            LOG(INFO) << "Successfully load lm file " << lm_file;
        }else{
            LOG(ERROR) << "Failed to load lm file " << lm_file;
@@ -309,6 +308,9 @@
{
    if(vocab){
        delete vocab;
    }
    if(lm_vocab){
        delete lm_vocab;
    }
    if(seg_dict){
        delete seg_dict;
@@ -687,6 +689,11 @@
    return vocab;
}
Vocab* Paraformer::GetLmVocab()
{
    return lm_vocab;
}
PhoneSet* Paraformer::GetPhoneSet()
{
    return phone_set_;