Переводчики Huggingface MarianMT теряют контент в зависимости от модели

Контекст

Я использую MarianMT von Huggingface через Python для перевода текста из источника на целевой язык.

Ожидаемое поведение

Я ввожу последовательность в модель MarianMT и получаю обратно эту последовательность. Для этого я использую соответствующую языковую модель и токенизатор. Все предложения, которые я ввожу, также возвращаются. Предложения рассматриваются как последовательность.

Текущее поведение

В зависимости от языковой модели модель не переводит все, а возвращает только части. В этом примере отсутствует последнее предложение:

Оригинал (немецкий): Ein Nilpferd Lief im Dschungel rum und musste aufs WC. Da traf es einen Kakadu und fragte nach dem Weg. Der sagte wenn du Kaka musst, dann pass mal ganz kurz auf. Ich sag dir wo du hanghen musst, ich kenn mich hier gut aus.

Результат (на английском языке): Бегемот бегал по джунглям и должен был сходить в туалет. Там был какаду и спросил дорогу. Он сказал, что если вам нужно Кака, то поосторожней минутку. Я скажу вам, куда вам нужно идти, я знаю дорогу здесь.

Результат (голландский): Een nijlpaard liep rond in de Jungle en moest naar het Toilet... en een kaketoe vroeg naar de weg... die zei dat als je Kaka moest, ik even moest oppassen.

Текущий код

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM


def translate_text(input, source, target):

    # Prepare output
    output = ""

    model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-" + source + "-" + target)
    tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-" + source + "-" + target)

    inputs = tokenizer.encode(input[:512], return_tensors = "pt", padding='longest')
    outputs = model.generate(inputs, max_length=4000, num_beams=4, early_stopping=True)

    for t in [tokenizer.convert_ids_to_tokens(s) for s in outputs.tolist()[0]]:
        output = output + t.replace("▁", " ").replace("</s>", "")

    output.replace("<pad>", "")

    return output


print(translate_text("Ein Nilpferd lief im Dschungel rum und musste aufs WC. Da traf es einen Kakadu und fragte nach dem Weg. Der sagte wenn du Kaka musst, dann pass mal ganz kurz auf. Ich sag dir wo du hingehen musst, ich kenn mich hier gut aus.", "de", "nl"))
print(translate_text("Ein Nilpferd lief im Dschungel rum und musste aufs WC. Da traf es einen Kakadu und fragte nach dem Weg. Der sagte wenn du Kaka musst, dann pass mal ganz kurz auf. Ich sag dir wo du hingehen musst, ich kenn mich hier gut aus.", "de", "en"))

Нужна помощь

Что мне не хватает? Почему некоторые части последовательности отсутствуют?