Tianlin Liu 0040469bb8 Correct attention mask dtype for Flax GPT2 (#25636) ...

* Correct attention mask dtype

* reformat code

* add a test for boolean mask

* convert test to fast test

* delete unwanted print

* use assertTrue for testing

2023-08-25 17:36:37 +02: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

CI with num_hidden_layers=2 🚀🚀🚀 (#25266)

2023-08-02 20:22:36 +02: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