riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

better naming for if statement

Browse Source
This commit is contained in:
patrickvonplaten 2019-12-23 23:39:16 +01:00
parent 7e0c5c731a
commit d039c679d2
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/modeling_utils.py
View File

@ -542,7 +542,11 @@ class PreTrainedModel(nn.Module):
def _do_output_past(self, outputs):
# TODO: might be better to write a self.do_output_past method for each individual class as is done for
# prepare_inputs_for_generation
if hasattr(self.config, 'output_past') and self.config.output_past and len(outputs) > 1 and not hasattr(self, 'mem_len'):
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')
if has_output_past and has_multiple_outputs and not has_mem_len:
return True
# TODO: Add cases for (xlnet, transfo_xl) using mem_len
return False