Skip to content

Commit cc853fc

Browse files
committed
fix beam search using wrong parent probability
1 parent 22a9104 commit cc853fc

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -732,13 +732,13 @@ struct LanguageModelState {
732732
index_value[i].first);
733733
}
734734

735-
if (sequences[i].first > 1.0f || sequences[i].first < 0.0f) {
735+
if (parent_seq.first > 1.0f || parent_seq.first < 0.0f) {
736736
AKLOGE("Expected sequences value to be probability [%.2f]",
737-
sequences[i].first);
737+
parent_seq.first);
738738
}
739739

740740
next_sequences.emplace_back(
741-
index_value[i].first * sequences[i].first,
741+
index_value[i].first * parent_seq.first,
742742
potential_sequence_data{
743743
new_sequence,
744744
parent_seq.second.seq_id

0 commit comments

Comments
 (0)