beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Revert "avoid sqrt computation for text encoder"

Browse source 
This reverts commit d40d28cfa2.
This commit is contained in:
0nullpointer 2024-01-09 16:19:30 +05:30
parent d40d28cfa2
commit ff0eaefd9b
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart
View file

@ -51,9 +51,8 @@ class OnnxTextEncoder {
textNormalization += embedding[i] * embedding[i];
}
final double sqrtTexteNormalization = sqrt(textNormalization);
for (int i = 0; i < 512; i++) {
embedding[i] = embedding[i] / sqrtTexteNormalization;
embedding[i] = embedding[i] / sqrt(textNormalization);
}
return (embedding);