Skip to content

Commit d7a69d3

Browse files
authored
Merge pull request #3364 from flairNLP/fix_embedding_size_for_xlm_roberta
fix embedding size for xlm roberta models
2 parents c375fba + 032260c commit d7a69d3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

flair/embeddings/transformer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1139,8 +1139,8 @@ def is_supported_t5_model(config: PretrainedConfig) -> bool:
11391139
# If we use a context separator, add a new special token
11401140
self.use_context_separator = use_context_separator
11411141
if use_context_separator:
1142-
self.tokenizer.add_special_tokens({"additional_special_tokens": [SENTENCE_BOUNDARY_TAG]})
1143-
transformer_model.resize_token_embeddings(len(self.tokenizer))
1142+
added = self.tokenizer.add_special_tokens({"additional_special_tokens": [SENTENCE_BOUNDARY_TAG]})
1143+
transformer_model.resize_token_embeddings(transformer_model.config.vocab_size + added)
11441144

11451145
super().__init__(**self.to_args())
11461146

0 commit comments

Comments
 (0)