Good word list in generate function

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