# Good word list in generate function

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgood-word-list-in-generate-function%2F10724%3E%5C
**Category:** 🤗Transformers\
**Created:** [October 12, 2021, 6:11pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgood-word-list-in-generate-function%2F10724 "2021-10-12T18:11:02Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![omidb](https://avatars.discourse-cdn.com/v4/letter/o/47e85d/32.png) [@omidb](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fomidb)%5C
**Post date:** [October 12, 2021, 6:11pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgood-word-list-in-generate-function%2F10724%2F1 "2021-10-12T18:11:03Z")

</div>

Hi,

I know it’s possible to pass `bad_words_ids` to the `generate` function. I was wondering what is the best way to do the opposite. Giving the model a word list and force it to only use them.

Thanks

---

<div class="post-metadata">

**Author:** ![ignacioiglesias](https://avatars.discourse-cdn.com/v4/letter/i/ac8455/32.png) [@ignacioiglesias](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fignacioiglesias)%5C
**Post date:** [March 23, 2023, 5:10pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgood-word-list-in-generate-function%2F10724%2F2 "2023-03-23T17:10:54Z")

</div>

Setting the logits to +Inf for certain token, i.e. opposite to what is done in with `bad_words_ids` would result in a sequence that only contains that token(s) and, probably, make no sense as a sentence. Something like [this](https://github.com/huggingface/transformers/issues/22168#issuecomment-1477998997) feature request I submitted would allow a bit more flexibility. Feel free to react/comment to it. Also check [this](https://huggingface.co/docs/transformers/internal/generation_utils#transformers.Constraint) from the docs.