Merge branch 'main' of https://github.com/ashvardanian/uform

ashvardanian · Apr 25, 2024 · f4b19a8 · f4b19a8
2 parents f195b66 + cf25160
commit f4b19a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/javascript/encoders.mjs b/javascript/encoders.mjs
@@ -50,8 +50,7 @@ class TextProcessor {
     async process(texts) {
 
         const encoded = await this.tokenizer(texts, {
-            addSpecialTokens: true,
-            returnAttentionMask: true,
+            add_special_tokens: true,
             padding: 'max_length',
             max_length: this.maxSeqLen,
             truncation: true,

diff --git a/javascript/encoders_test.js b/javascript/encoders_test.js
@@ -101,7 +101,7 @@ async function tryImageEncoderForwardPass(modelId) {
 }
 
 function cosineSimilarity(vecA, vecB) {
-    // We may be receiving a complex tesnor type, so let's check if it
+    // We may be receiving a complex tensor type, so let's check if it
     // has an array member named `data`.
     if (vecA.data) {
         vecA = vecA.data;
@@ -180,8 +180,8 @@ async function tryCrossReferencingImageAndText(modelId) {
         const textEmbedding = await textEncoder.encode(processedText);
         const imageEmbedding = await imageEncoder.encode(processedImage);
 
-        textEmbeddings.push(new Float32Array(textEmbedding.embeddings.cpuData));
-        imageEmbeddings.push(new Float32Array(imageEmbedding.embeddings.cpuData));
+        textEmbeddings.push(new Float32Array(textEmbedding.embeddings.data));
+        imageEmbeddings.push(new Float32Array(imageEmbedding.embeddings.data));
 
         // Print-based debugging at its best :)
         // console.log(`Text: ${text}, Image: ${imageUrl}`);