| | |
|---|
| | | */ |
|---|
| | | #pragma once |
|---|
| | | |
|---|
| | | #ifndef PARAFORMER_MODELIMP_H |
|---|
| | | #define PARAFORMER_MODELIMP_H |
|---|
| | | |
|---|
| | | #include "precomp.h" |
|---|
| | | |
|---|
| | | namespace paraformer { |
|---|
| | | namespace funasr { |
|---|
| | | |
|---|
| | | class Paraformer : public Model { |
|---|
| | | /** |
|---|
| | |
|---|
| | | //std::unique_ptr<knf::OnlineFbank> fbank_; |
|---|
| | | knf::FbankOptions fbank_opts; |
|---|
| | | |
|---|
| | | Vocab* vocab; |
|---|
| | | Vocab* vocab = nullptr; |
|---|
| | | vector<float> means_list; |
|---|
| | | vector<float> vars_list; |
|---|
| | | const float scale = 22.6274169979695; |
|---|
| | |
|---|
| | | void ApplyCmvn(vector<float> *v); |
|---|
| | | string GreedySearch( float* in, int n_len, int64_t token_nums); |
|---|
| | | |
|---|
| | | std::shared_ptr<Ort::Session> m_session; |
|---|
| | | std::shared_ptr<Ort::Session> m_session = nullptr; |
|---|
| | | Ort::Env env_; |
|---|
| | | Ort::SessionOptions session_options; |
|---|
| | | |
|---|
| | |
|---|
| | | string Rescoring(); |
|---|
| | | }; |
|---|
| | | |
|---|
| | | } // namespace paraformer |
|---|
| | | #endif |
|---|
| | | } // namespace funasr |
|---|
