We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents c375fba + 032260c commit d7a69d3Copy full SHA for d7a69d3
flair/embeddings/transformer.py
@@ -1139,8 +1139,8 @@ def is_supported_t5_model(config: PretrainedConfig) -> bool:
1139
# If we use a context separator, add a new special token
1140
self.use_context_separator = use_context_separator
1141
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))
+ added = self.tokenizer.add_special_tokens({"additional_special_tokens": [SENTENCE_BOUNDARY_TAG]})
+ transformer_model.resize_token_embeddings(transformer_model.config.vocab_size + added)
1144
1145
super().__init__(**self.to_args())
1146
0 commit comments