Set normalize_embeddings to True
Browse files- modeling_xlm_roberta.py +1 -1
modeling_xlm_roberta.py
CHANGED
|
@@ -470,7 +470,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
|
|
| 470 |
convert_to_numpy: bool = True,
|
| 471 |
convert_to_tensor: bool = False,
|
| 472 |
device: Optional[torch.device] = None,
|
| 473 |
-
normalize_embeddings: bool =
|
| 474 |
truncate_dim: Optional[int] = None,
|
| 475 |
adapter_mask: Optional[torch.Tensor] = None,
|
| 476 |
task_type: Optional[str] = None,
|
|
|
|
| 470 |
convert_to_numpy: bool = True,
|
| 471 |
convert_to_tensor: bool = False,
|
| 472 |
device: Optional[torch.device] = None,
|
| 473 |
+
normalize_embeddings: bool = True,
|
| 474 |
truncate_dim: Optional[int] = None,
|
| 475 |
adapter_mask: Optional[torch.Tensor] = None,
|
| 476 |
task_type: Optional[str] = None,
|