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