tests/test_modeling_common.py · 5bbf6712767a84ac4c4de1a5525a1b4e697064f7 · chenpangpang / transformers

[`core`] Refactor of `gradient_checkpointing` (#27020) · 06e782da

Younes Belkada authored Oct 25, 2023

* v1

* fix

* remove `create_custom_forward`

* fixup

* fixup

* add test and fix all failing GC tests

* remove all remaining `create_custom_forward` methods

* fix idefics bug

* fixup

* replace with `__call__`

* add comment

* quality

06e782da

test_modeling_common.py 137 KB

Replace test_modeling_common.py