jupyterjazz commited on
Commit
f2933ac
1 Parent(s): c0215f6

Set normalize_embeddings to True

Browse files
Files changed (1) hide show
  1. 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 = False,
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,