fix #2214

8efc6dd5 · Lysandre · 94c99db3 · 8efc6dd5
Commit 8efc6dd5 authored Dec 18, 2019 by Lysandre
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

transformers/configuration_xlm.py transformers/configuration_xlm.py +3 -0

No files found.
--- a/transformers/configuration_xlm.py
+++ b/transformers/configuration_xlm.py
@@ -144,6 +144,9 @@ class XLMConfig(PretrainedConfig):
        self.start_n_top = start_n_top
        self.end_n_top = end_n_top

+        if "n_words" in kwargs:
+            self.n_words = kwargs["n_words"]
+
    @property
    def n_words(self):  # For backward compatibility
        return self.vocab_size