make lint

embeddings-benchmark · Nov 9, 2024 · fcc76d4 · fcc76d4
1 parent da1a189
commit fcc76d4
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/mteb/models/vlm2vec_models.py b/mteb/models/vlm2vec_models.py
@@ -225,17 +225,15 @@ def get_text_embeddings(self, texts: list[str], batch_size: int = 32):
 
         all_text_embeddings = torch.cat(all_text_embeddings, dim=0)
         return all_text_embeddings
-    
+
     def calculate_probs(self, text_embeddings, image_embeddings):
-            text_embeddings = text_embeddings / text_embeddings.norm(
-                dim=-1, keepdim=True
-            )
-            image_embeddings = image_embeddings / image_embeddings.norm(
-                dim=-1, keepdim=True
-            )
-            logits = torch.matmul(image_embeddings, text_embeddings.T)
-            probs = (logits * 100).softmax(dim=-1)
-            return probs
+        text_embeddings = text_embeddings / text_embeddings.norm(dim=-1, keepdim=True)
+        image_embeddings = image_embeddings / image_embeddings.norm(
+            dim=-1, keepdim=True
+        )
+        logits = torch.matmul(image_embeddings, text_embeddings.T)
+        probs = (logits * 100).softmax(dim=-1)
+        return probs
 
     def get_fused_embeddings(
         self,