jupyterjazz commited on
Commit
b27fa55
1 Parent(s): fd34c40

fix: another one

Browse files

Signed-off-by: jupyterjazz <saba.sturua@jina.ai>

Files changed (1) hide show
  1. modeling_xlm_roberta.py +1 -1
modeling_xlm_roberta.py CHANGED
@@ -579,7 +579,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
579
  all_embeddings = [all_embeddings[idx] for idx in inverse_permutation]
580
 
581
  if truncate_dim:
582
- if not self.config.matryoshka_dimension:
583
  logger.warning(
584
  'Matryoshka embeddings are not supported, so dimension truncation will not be performed.'
585
  )
 
579
  all_embeddings = [all_embeddings[idx] for idx in inverse_permutation]
580
 
581
  if truncate_dim:
582
+ if not self.config.matryoshka_dimensions:
583
  logger.warning(
584
  'Matryoshka embeddings are not supported, so dimension truncation will not be performed.'
585
  )