tests/test_modeling_common.py · 2e7e4280aa6f380a4e3afad6524295a17901c56c · chenpangpang / transformers

Traced models serialization and torchscripting fix (#17206) · 2e7e4280

Michael Benayoun authored May 23, 2022

* Fix torch.jit.script and pickling issues

* Fix get_attr issues

* Fix import in function

* Fix GPT-J and T5 tracing for torch=1.11

* Gate graph surgery on torch version

* Modeling minor changes to enable TorchScripting

* Model serialization / deserialization test

* Remove _assert_is_none users

2e7e4280

test_modeling_common.py 120 KB

Replace test_modeling_common.py