跳到正文
Hugging Face Blog·· 2022-03-11AI 评分38

在 🤗 Transformers 中用约束束搜索引导文本生成

Guiding Text Generation with Constrained Beam Search in 🤗 Transformers

AI 导读

Hugging Face 在 🤗 Transformers 中推出约束束搜索(constrained beam search),通过 model.generate() 的 force_words_ids 参数让用户指定生成结果中必须包含的词。该功能支持强制单个词、多个短语以及析取约束(在候选词列表中至少包含一个),可在生成阶段直接控制输出,无需事后再筛选。

来源:Hugging Face Blog · huggingface.co