Skip to content

GitLab

Commit 8efc6dd5 authored by Lysandre's avatar Lysandre
Browse files

fix #2214

parent 94c99db3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
transformers/configuration_xlm.py
View file @ 8efc6dd5
...@@ -144,6 +144,9 @@ class XLMConfig(PretrainedConfig): ...@@ -144,6 +144,9 @@ class XLMConfig(PretrainedConfig):
self.start_n_top = start_n_top self.start_n_top = start_n_top
self.end_n_top = end_n_top self.end_n_top = end_n_top
if "n_words" in kwargs:
self.n_words = kwargs["n_words"]
@property @property
def n_words(self): # For backward compatibility def n_words(self): # For backward compatibility
return self.vocab_size return self.vocab_size
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment