transformers/tests/models/gpt2
Younes Belkada 285a48011d
Fix gradient checkpointing + fp16 autocast for most models (#24247)
* fix gc bug

* continue PoC on OPT

* fixes

* 🤯

* fix tests

* remove pytest.mark

* fixup

* forward contrib credits from discussions

* forward contrib credits from discussions

* reverting changes on untouched files.

---------

Co-authored-by: zhaoqf123 <zhaoqf123@users.noreply.github.com>
Co-authored-by: 7eu7d7 <7eu7d7@users.noreply.github.com>
2023-06-21 17:04:59 +02: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 gradient checkpointing + fp16 autocast for most models (#24247) 2023-06-21 17:04:59 +02:00
test_modeling_tf_gpt2.py TF: standardize test_model_common_attributes for language models (#23457) 2023-06-13 17:51:37 +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