Files changed (1) hide show
  1. modeling_clip.py +2 -2
modeling_clip.py CHANGED
@@ -289,7 +289,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
289
  convert_to_numpy: bool = True,
290
  convert_to_tensor: bool = False,
291
  device: Optional[torch.device] = None,
292
- normalize_embeddings: bool = False,
293
  **tokenizer_kwargs,
294
  ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
295
  """
@@ -405,7 +405,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
405
  convert_to_numpy: bool = True,
406
  convert_to_tensor: bool = False,
407
  device: Optional[torch.device] = None,
408
- normalize_embeddings: bool = False,
409
  ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
410
  """
411
  Computes image embeddings.
 
289
  convert_to_numpy: bool = True,
290
  convert_to_tensor: bool = False,
291
  device: Optional[torch.device] = None,
292
+ normalize_embeddings: bool = True,
293
  **tokenizer_kwargs,
294
  ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
295
  """
 
405
  convert_to_numpy: bool = True,
406
  convert_to_tensor: bool = False,
407
  device: Optional[torch.device] = None,
408
+ normalize_embeddings: bool = True,
409
  ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
410
  """
411
  Computes image embeddings.