LLamaQaStoppingCriteria not in transformers.generation.stopping_criteria?

[li-yang23]

trafficstars

I just checked codes in generation.py and I noticed it import LLamaQaStoppingCriteria from transformers.generation.stopping_criteria. However I did not find this function (or class) in the transformers/generation/stopping_criteria.py script. (not from the transformers installed by pip nor in this repo), may be it comes from another dependency?

[voidism]

Hi!

It's from this part of the code:

https://github.com/voidism/Lookback-Lens/blob/8298a9525943b3398fe20316867a53b322647c98/transformers-4.32.0/src/transformers/generation/stopping_criteria.py#L43

If you pip install ./transformers-4.32.0, you should be able to import it, at least it worked in my test. Let my know if it's still not working!