Merge pull request #3364 from flairNLP/fix_embedding_size_for_xlm_rob…

…erta fix embedding size for xlm roberta models
flairNLP · Oct 28, 2023 · d7a69d3 · d7a69d3
2 parents c375fba + 032260c
commit d7a69d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flair/embeddings/transformer.py b/flair/embeddings/transformer.py
@@ -1139,8 +1139,8 @@ def is_supported_t5_model(config: PretrainedConfig) -> bool:
         # If we use a context separator, add a new special token
         self.use_context_separator = use_context_separator
         if use_context_separator:
-            self.tokenizer.add_special_tokens({"additional_special_tokens": [SENTENCE_BOUNDARY_TAG]})
-            transformer_model.resize_token_embeddings(len(self.tokenizer))
+            added = self.tokenizer.add_special_tokens({"additional_special_tokens": [SENTENCE_BOUNDARY_TAG]})
+            transformer_model.resize_token_embeddings(transformer_model.config.vocab_size + added)
 
         super().__init__(**self.to_args())