src/transformers/generation/stopping_criteria.py · c1aa0edb48217f416f4bbe6e3a9db1500284513b · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[generate] only require an attention mask for mps with torch<2.4 (#32367) · c1aa0edb
Sanchit Gandhi authored Aug 02, 2024
```
* up

* style

* stopping
```
c1aa0edb

stopping_criteria.py 28.7 KB