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

avoid sqrt computation for text encoder

Browse source
This commit is contained in:
0nullpointer 2024-01-09 16:18:00 +05:30
parent e21b862f53
commit d40d28cfa2
1 changed files with 2 additions and 1 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,8 +51,9 @@ class OnnxTextEncoder {
textNormalization += embedding[i] * embedding[i]; textNormalization += embedding[i] * embedding[i];
} }
final double sqrtTexteNormalization = sqrt(textNormalization);
for (int i = 0; i < 512; i++) { for (int i = 0; i < 512; i++) {
embedding[i] = embedding[i] / sqrt(textNormalization); embedding[i] = embedding[i] / sqrtTexteNormalization;
} }
return (embedding); return (embedding);