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	Add Switch transformers (#19323)	2022-11-15 13:06:45 +01:00
test_modeling_switch_transformers.py	Fix gradient checkpointing + fp16 autocast for most models (#24247)	2023-06-21 17:04:59 +02:00