refactor: optional arg

Signed-off-by: jupyterjazz <[email protected]>

modeling_xlm_roberta.py

@@ -452,7 +452,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
         convert_to_tensor: bool = False,
         device: Optional[torch.device] = None,
         normalize_embeddings: bool = False,
-        truncate_dim: int = None,
+        truncate_dim: Optional[int] = None,
         **tokenizer_kwargs,
     ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]:
         """