Easy-Transformer
Easy-Transformer copied to clipboard
[Feature Request] Add Stopping Criteria support
Hey,
Huggingface offers a useful Stopping Criteria class that can be utilized in the .generate function to halt generation based on custom conditions.
Do you think it would be possible to integrate this functionality into HookedTransformer.generate as well?
I think this could be a very useful thing to add
Hi @bryce13950 , if this isn't being actively worked on, I'd like to attempt this.