EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Fix context graph (#292)

Browse Source
This commit is contained in:
Wei Kang
2023-08-28 19:39:22 +08:00
committed by GitHub
parent 49ec7e8f57
commit 2b0152d2a2
3 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sherpa-onnx/csrc/context-graph.h
View File

@@ -21,7 +21,7 @@ struct ContextState {
int32_t token;
float token_score;
float node_score;
float local_node_score;
float output_score;
bool is_end;
std::unordered_map<int32_t, std::unique_ptr<ContextState>> next;
const ContextState *fail = nullptr;
@@ -29,11 +29,11 @@ struct ContextState {
ContextState() = default;
ContextState(int32_t token, float token_score, float node_score,
float local_node_score, bool is_end)
float output_score, bool is_end)
: token(token),
token_score(token_score),
node_score(node_score),
local_node_score(local_node_score),
output_score(output_score),
is_end(is_end) {}
};