jinaai
/

jina-clip-implementation

🇪🇺 Region: EU

Model card Files Files and versions Community

koukandre commited on Jun 5

Commit

b60cb6d

•

1 Parent(s): 92ddc0b

Update modeling_clip.py

Files changed (1) hide show

modeling_clip.py +2 -2

modeling_clip.py CHANGED Viewed

@@ -289,7 +289,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
         convert_to_numpy: bool = True,
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
-        normalize_embeddings: bool = False,
         **tokenizer_kwargs,
     ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
         """
@@ -405,7 +405,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
         convert_to_numpy: bool = True,
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
-        normalize_embeddings: bool = False,
     ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
         """
         Computes image embeddings.

         convert_to_numpy: bool = True,
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
+        normalize_embeddings: bool = True,
         **tokenizer_kwargs,
     ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
         """
         convert_to_numpy: bool = True,
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
+        normalize_embeddings: bool = True,
     ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
         """
         Computes image embeddings.