jinaai
/

jina-clip-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

Update modeling_clip.py

#5

by koukandre - opened Jun 5

base: refs/heads/main

←

from: refs/pr/5

Discussion Files changed

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.