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

🤗 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