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