python - 使用谷歌的 T5 将德语翻译成英语

Question

我正在尝试使用 Google 的 T5 进行语言翻译。但是，它不适用于德语到英语。

英语到德语工作正常：

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])

但是，将编码更改为“德语到英语”不起作用。

这个模型是不是不能把德语翻译成英语，还是我用错了？

标签： pythonnlphuggingface-transformerslanguage-translation