From d997ac78105278bff96d94115b505f7ab84d4fcf Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 10 Mar 2020 20:24:26 +0100 Subject: [PATCH] fix typo --- 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 82434100c4..62c6f9950d 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -1281,7 +1281,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin): if self.config.is_encoder_decoder: # do not return first token return decoded[:, 1:] -# return decoded + return decoded # force one of token_ids to be generated by setting prob of all other tokens to 0. def _force_token_ids_generation(self, scores, token_ids):