From eeaa402cd4b3cd84e525fbdd158d525650e3a47f Mon Sep 17 00:00:00 2001 From: patrickvonplaten Date: Mon, 23 Dec 2019 22:15:06 +0100 Subject: [PATCH] rename comments --- src/transformers/modeling_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 54283617e0..9a07e56368 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -731,7 +731,7 @@ class PreTrainedModel(nn.Module): outputs = self(**model_inputs) next_token_logits = outputs[0][:, -1, :] - # if model has past, then set the past parameter to speed up decoding + # if model has past, then set the past variable to speed up decoding if self._has_past(outputs): past = outputs[1] @@ -818,7 +818,7 @@ class PreTrainedModel(nn.Module): outputs = self(**model_inputs) # (batch_size * num_beams, cur_len, vocab_size) scores = outputs[0][:, -1, :] # (batch_size * num_beams, vocab_size) - # if model has past, then set the past parameter to speed up decoding + # if model has past, then set the past variable to speed up decoding if self._has_past(outputs): past = outputs[1]