jupyterjazz
commited on
Commit
•
b27fa55
1
Parent(s):
fd34c40
fix: another one
Browse filesSigned-off-by: jupyterjazz <saba.sturua@jina.ai>
- 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.
|
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 |
)
|