fix argument order (#3637)
This commit is contained in:
committed by
GitHub
parent
b809d2f073
commit
1789c7daf1
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user