From 6bca56fdb0587a4291f8465a0a6e818f5541a5e3 Mon Sep 17 00:00:00 2001 From: patrickvonplaten Date: Tue, 24 Dec 2019 01:02:58 +0100 Subject: [PATCH] check for self.config.mem_len instead of self.mem_len in _do_output_past --- src/transformers/modeling_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 3248763bdb..f81bcbecae 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -544,7 +544,7 @@ class PreTrainedModel(nn.Module): # individual class as is done for prepare_inputs_for_generation has_output_past = hasattr(self.config, 'output_past') and self.config.output_past has_multiple_outputs = len(outputs) > 1 - has_mem_len = hasattr(self, 'mem_len') + has_mem_len = hasattr(self.config, 'mem_len') if has_output_past and has_multiple_outputs and not has_mem_len: return True