transformers/tests/models/gpt2
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_flax_gpt2.py Correct attention mask dtype for Flax GPT2 (#25636) 2023-08-25 17:36:37 +02:00
test_modeling_gpt2.py device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
test_modeling_tf_gpt2.py Speed up TF tests by reducing hidden layer counts (#24595) 2023-06-30 16:30:33 +01:00
test_tokenization_gpt2_tf.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_tokenization_gpt2.py 🚨🚨 🚨🚨 [Tokenizer] attemp to fix add_token issues🚨🚨 🚨🚨 (#23909) 2023-09-18 20:28:36 +02:00