transformers/tests/models/gpt2
Matt 134caef31a
Speed up TF tests by reducing hidden layer counts (#24595)
* hidden layers, huh, what are they good for (absolutely nothing)

* Some tests break with 1 hidden layer, use 2

* Use 1 hidden layer in a few slow models

* Use num_hidden_layers=2 everywhere

* Slightly higher tol for groupvit

* Slightly higher tol for groupvit
2023-06-30 16:30:33 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_gpt2.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_gpt2.py Fix TypeError: Object of type int64 is not JSON serializable (#24340) 2023-06-27 12:15:49 +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 [OPT] Adds GPT2TokenizerFast to the list of tokenizer to use for OPT. (#20823) 2023-02-07 17:35:28 +01:00