adapt style to predefined style layout

2019-12-25 23:32:44 +01:00
parent deff792bb6
commit fc84bd5254
5 changed files with 8 additions and 8 deletions
--- a/src/transformers/modeling_transfo_xl.py
+++ b/src/transformers/modeling_transfo_xl.py
@@ -935,7 +935,7 @@ class TransfoXLLMHeadModel(TransfoXLPreTrainedModel):
        inputs = {"input_ids": input_ids}

        # if past is defined in model kwargs then use it for faster decoding
-        if 'past' in model_kwargs and model_kwargs['past']:
-            inputs['mems'] = model_kwargs['past']
+        if "past" in model_kwargs and model_kwargs["past"]:
+            inputs["mems"] = model_kwargs["past"]

        return inputs