transformers/tests/models/rwkv
Joao Gante 1fd60fec75
RWKV: enable generation tests (#31490)
* add rwkv tests

* has_attentions set in individual tests
2024-06-20 14:15:01 +01:00
..
__init__.py Add RWKV-4 (#22797) 2023-05-09 13:04:10 -04:00
test_modeling_rwkv.py RWKV: enable generation tests (#31490) 2024-06-20 14:15:01 +01:00