Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "fbd746bd065a9aaacd1ef25840cdc9ec957e8cac"
Unverified Commit 1789c7da authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix argument order (#3637)

parent b809d2f0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_tf_transfo_xl.py
View file @ 1789c7da
...@@ -488,7 +488,7 @@ class TFTransfoXLMainLayer(tf.keras.layers.Layer): ...@@ -488,7 +488,7 @@ class TFTransfoXLMainLayer(tf.keras.layers.Layer):
else: else:
return None return None
def _update_mems(self, hids, mems, qlen, mlen): def _update_mems(self, hids, mems, mlen, qlen):
# does not deal with None # does not deal with None
if mems is None: if mems is None:
return None return None
......
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