fix argument order (#3637)

1789c7da · Patrick von Platen · GitHub · b809d2f0 · 1789c7da
Unverified Commit 1789c7da authored Apr 05, 2020 by Patrick von Platen Committed by GitHub Apr 05, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/src/transformers/modeling_tf_transfo_xl.py
+++ b/src/transformers/modeling_tf_transfo_xl.py
@@ -488,7 +488,7 @@ class TFTransfoXLMainLayer(tf.keras.layers.Layer):
        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