GH-1492: added new BERT embeddings implementation, with original name

flairNLP · Mar 26, 2020 · 4a9e629 · 4a9e629
1 parent 5a081a2
commit 4a9e629
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flair/embeddings.py b/flair/embeddings.py
@@ -2198,7 +2198,7 @@ def __setstate__(self, d):
                 self.word_embeddings[key] = self.word_embeddings[key].cpu()
 
 
-class BertEmbeddingsNew(DocumentEmbeddings, TokenEmbeddings):
+class BertEmbeddings(DocumentEmbeddings, TokenEmbeddings):
     def __init__(
         self,
         bert_model_or_path: str = "bert-base-uncased",