def METHOD_NAME(self, bos_token_id, encoder_outputs):
    input_ids = org_prepare_input_ids_for_generation(self, bos_token_id, encoder_outputs)
    if input_ids.size()[0] != curr_batch_size[0]:
        input_ids = input_ids.repeat(curr_batch_size[0], 1)
    return input_ids