tests/models/rwkv/test_modeling_rwkv.py · 1fd60fec75b2f75432cca48ff79d606e084a9bc2 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

RWKV: enable generation tests (#31490) · 1fd60fec
Joao Gante authored Jun 20, 2024
```
* add rwkv tests

* has_attentions set in individual tests
```
1fd60fec

test_modeling_rwkv.py 19.3 KB