fix _update_memory fn call in transformer-xl (#2971)

ee60840e · Sandro Cavallari · GitHub · 6a50d501 · ee60840e
Unverified Commit ee60840e authored Feb 25, 2020 by Sandro Cavallari Committed by GitHub Feb 24, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_transfo_xl.py src/transformers/modeling_transfo_xl.py +1 -1

No files found.
--- a/src/transformers/modeling_transfo_xl.py
+++ b/src/transformers/modeling_transfo_xl.py
@@ -645,7 +645,7 @@ class TransfoXLModel(TransfoXLPreTrainedModel):
        else:
            return None

-    def _update_mems(self, hids, mems, qlen, mlen):
+    def _update_mems(self, hids, mems, mlen, qlen):
        # does not deal with None
        if mems is None:
            return None