BART/TFBart: allow decoder_input_ids.shape[-1] > 1 + use_cache = True
There is an edge case fixed in https://github.com/huggingface/transformers/pull/4581
Try to apply the same logic to bart/move input_id slicing to prepare_inputs_for_generation