How to set stopping criteria in model.generate() when a certain word appears

🤗Transformers
4

@Pradeep1995 this may be a good solution.

1 Like