def METHOD_NAME(
    self,
    tokenized_sentences: List[torch.LongTensor],
    *args,
    inference_step_args=None,
    **kwargs
) -> List[List[Dict[str, torch.Tensor]]]:
    inference_step_args = inference_step_args or {}
    if "prefix_tokens" in inference_step_args:
        raise NotImplementedError("prefix generation not implemented for BART")
    else:
        bsz = len(tokenized_sentences)
        inference_step_args["prefix_tokens"] = tokenized_sentences[0].new_full(
            (bsz, 1), fill_value=self.task.source_dictionary.bos()
        ).to(device=self.device)
    return super().METHOD_NAME(
        tokenized_sentences,
        *args,
        inference_step_args=inference_step_args,
        **kwargs
    )