transformers/tests/models/roformer
TheWall9 ad5e9b6c6a
[Roformer] Fixing a bug in RoFormerEncoder where it was ignoring the length of past_key_values when generating as a decoder (#22416)
* fix RoFormerEncoder postion embedding when generate as decoder

* make fixup

* add test case for check generate with past key values

* remove duplicating code
2023-04-04 12:50:33 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_roformer.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_roformer.py [Roformer] Fixing a bug in RoFormerEncoder where it was ignoring the length of past_key_values when generating as a decoder (#22416) 2023-04-04 12:50:33 +02:00
test_modeling_tf_roformer.py Move is_pipeline_test_to_skip to specific model test classes (#21999) 2023-03-14 10:03:02 +01:00
test_tokenization_roformer.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00