I am trying to use Google's T5 for language translation. However, it is not working for German to English.
English to German works fine:
self.tokenizer = AutoTokenizer.from_pretrained("t5-small")
self.model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
inputs = self.tokenizer.encode("translate English to German: " + text, return_tensors="pt", max_length=512, truncation=True)
summary_ids = self.model.generate(inputs, max_length=512, min_length=5, length_penalty=5., num_beams=2)
summary = self.tokenizer.decode(summary_ids[0])
However, changing encoding to "German to English" is not working.
Is this model not intended to be able to translate German to English, or am I using it wrong?