jinaai
/

xlm-roberta-flash-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

jupyterjazz commited on Sep 11

Commit

f2933ac

•

1 Parent(s): c0215f6

Set normalize_embeddings to True

Files changed (1) hide show

modeling_xlm_roberta.py +1 -1

modeling_xlm_roberta.py CHANGED Viewed

@@ -470,7 +470,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
         convert_to_numpy: bool = True,
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
-        normalize_embeddings: bool = False,
         truncate_dim: Optional[int] = None,
         adapter_mask: Optional[torch.Tensor] = None,
         task_type: Optional[str] = None,

         convert_to_numpy: bool = True,
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
+        normalize_embeddings: bool = True,
         truncate_dim: Optional[int] = None,
         adapter_mask: Optional[torch.Tensor] = None,
         task_type: Optional[str] = None,