transformers/tests/models/blenderbot
Hz, Ji 50378cbf6c
device agnostic models testing (#27146)
* device agnostic models testing

* add decorator `require_torch_fp16`

* make style

* apply review suggestion

* Oops, the fp16 decorator was misused
2023-10-31 18:12:14 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_blenderbot.py device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
test_modeling_flax_blenderbot.py Shorten the conversation tests for speed + fixing position overflows (#26960) 2023-10-31 14:20:04 +00:00
test_modeling_tf_blenderbot.py Shorten the conversation tests for speed + fixing position overflows (#26960) 2023-10-31 14:20:04 +00:00
test_tokenization_blenderbot.py Overhaul Conversation class and prompt templating (#25323) 2023-09-14 15:10:34 +01:00