Michael Benayoun 2e7e4280aa Traced models serialization and torchscripting fix (#17206) ... * 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		2022-05-23 17:50:40 +02:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_swin.py	Traced models serialization and torchscripting fix (#17206)	2022-05-23 17:50:40 +02:00
test_modeling_tf_swin.py	Add Tensorflow Swin model (#16988)	2022-05-16 22:19:53 +01:00