Update modeling_clip.py
Browse files- 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 =
|
| 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 =
|
| 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.
|