jinaai
/

xlm-roberta-flash-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

jupyterjazz commited on May 15

Commit

b27fa55

•

1 Parent(s): fd34c40

fix: another one

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

Files changed (1) hide show

modeling_xlm_roberta.py +1 -1

modeling_xlm_roberta.py CHANGED Viewed

@@ -579,7 +579,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
  all_embeddings = [all_embeddings[idx] for idx in inverse_permutation]
  if truncate_dim:
- if not self.config.matryoshka_dimension:
  logger.warning(
  'Matryoshka embeddings are not supported, so dimension truncation will not be performed.'
  )

  all_embeddings = [all_embeddings[idx] for idx in inverse_permutation]
  if truncate_dim:
+ if not self.config.matryoshka_dimensions:
  logger.warning(
  'Matryoshka embeddings are not supported, so dimension truncation will not be performed.'
  )