Skip to content

GitLab

Unverified Commit ee60840e authored by Sandro Cavallari's avatar Sandro Cavallari Committed by GitHub
Browse files

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

parent 6a50d501
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_transfo_xl.py
View file @ ee60840e
...@@ -645,7 +645,7 @@ class TransfoXLModel(TransfoXLPreTrainedModel): ...@@ -645,7 +645,7 @@ class TransfoXLModel(TransfoXLPreTrainedModel):
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