n/futokb
1
0
Fork 0
mirror of https://gitlab.futo.org/keyboard/latinime.git synced 2024-09-19 23:28:31 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Skip invalid token mixes

Browse Source
This commit is contained in:
Aleksandras Kostarevas 2024-05-16 12:18:13 -05:00
parent edbb92114f
commit e19de589f1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp
View File

@ -393,7 +393,8 @@ struct LanguageModelState {
// mix_f[0], mix_f[1], mix_f[2], mix_f[3], mix_f[4], mix_f[5], mix_f[6]);
} else {
for (auto &t: mix.mixes) {
if (t.weight < EPS) break;
if (t.weight < EPS) continue;
if (t.token < 0 || t.token >= (int)n_vocab) continue;
float *src = ((LlamaAdapter *) model->adapter)->embeddings.data() +
(t.token * n_embd);