From c3317e1f80e0b003fafda4e616cf8ccbeea95b2d Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 7 Sep 2020 02:16:24 -0700 Subject: [PATCH] typo (#6959) there is no var `decoder_input_ids`, but there is `input_ids` for decoder :) --- src/transformers/generation_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/generation_utils.py b/src/transformers/generation_utils.py index 638bb3b12e..baa2620265 100644 --- a/src/transformers/generation_utils.py +++ b/src/transformers/generation_utils.py @@ -411,7 +411,7 @@ class GenerationMixin: ) # shape: (batch_size * num_return_sequences * num_beams, cur_len) if self.config.is_encoder_decoder: - # create empty decoder_input_ids + # create empty decoder input_ids input_ids = torch.full( (effective_batch_size * num_beams, 1), decoder_start_token_id,