首页 > 解决方案 > 使用 MT5 生成时添加启动令牌

问题描述

我有一个使用huggingface提供的 mT5 模型生成文本的特定用例。我需要能够控制生成文本的语言。我尝试为编码器输入添加前缀,但模型总是生成英文文本。有没有办法为解码器输入添加前缀,以便我可以使用解码器前缀来控制生成文本的语言。我正在使用model.generate()函数来生成文本。

标签: pythonpytorchmultilingualhuggingface-transformersnlg

解决方案


推荐阅读