Hugging Face Blog·· 2022-03-11AI 评分46
在 🤗 Transformers 中用约束束搜索引导文本生成
Guiding Text Generation with Constrained Beam Search in 🤗 Transformers
AI 导读
Hugging Face 为 🤗 Transformers 的 model.generate() 新增约束束搜索（constrained beam search）功能，通过 force_words_ids 参数强制生成结果包含指定词。
来源：Hugging Face Blog · huggingface.co