Merge pull request #3229 from flairNLP/fix_loading_certain_models

Fix loading of (not so) old models
flairNLP · Apr 29, 2023 · fb023c7 · fb023c7
2 parents 4435a31 + 72efbb1
commit fb023c7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flair/embeddings/transformer.py b/flair/embeddings/transformer.py
@@ -1172,6 +1172,10 @@ def __setstate__(self, state):
 
     @classmethod
     def from_params(cls, params):
+        params.pop("truncate", None)
+        params.pop("stride", None)
+        params.pop("embedding_length", None)
+        params.pop("use_lang_emb", None)
         params["use_context"] = params.pop("context_length", 0)
         config_state_dict = params.pop("config_state_dict", None)
         config = None