tests/generation/test_stopping_criteria.py · 28316d0e8b4726c39bf75572c7788792c7da9ce9 · chenpangpang / transformers

Fix single letter stop strings (#31448) · 28316d0e

Matt authored Jun 18, 2024



* Fix single letter stop strings

* Change the 0 to a 1 to avoid potential empty vector headaches later

* Restructure for clarity

* Update tests/generation/test_stopping_criteria.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Add the unsqueeze

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

28316d0e

test_stopping_criteria.py 11.1 KB

Replace test_stopping_criteria.py