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
..
benchmark	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
deepspeed	missing file (#17164)	2022-05-10 10:19:50 -07:00
extended	Update self-push workflow (#17177)	2022-05-13 16:28:00 +02:00
fixtures	add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629)	2022-04-11 11:06:10 +02:00
generation	[Generation] Fix Transition probs (#17311)	2022-05-19 22:17:02 +02:00
models	Traced models serialization and torchscripting fix (#17206)	2022-05-23 17:50:40 +02:00
onnx	Add onnx export cuda support (#17183)	2022-05-18 17:52:13 +02:00
optimization	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
pipelines	Adding batch_size test to QA pipeline. (#17330)	2022-05-19 14:28:12 -04:00
sagemaker	Black preview (#17217)	2022-05-12 16:25:55 -04:00
tokenization	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
trainer	[tests] fix copy-n-paste error (#17312)	2022-05-18 16:00:47 -07:00
utils	Fix tests of mixed precision now that experimental is deprecated (#17300)	2022-05-17 14:14:17 +01:00
__init__.py
test_configuration_common.py	Black preview (#17217)	2022-05-12 16:25:55 -04:00
test_feature_extraction_common.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_common.py	Traced models serialization and torchscripting fix (#17206)	2022-05-23 17:50:40 +02:00
test_modeling_flax_common.py	[Flax] improve large model init and loading (#16148)	2022-04-19 14:19:55 +02:00
test_modeling_tf_common.py	Rewrite TensorFlow train_step and test_step (#17057)	2022-05-17 14:36:23 +01:00
test_sequence_feature_extraction_common.py	Some tests misusing assertTrue for comparisons fix (#16771)	2022-04-19 14:44:08 +02:00
test_tokenization_common.py	Black preview (#17217)	2022-05-12 16:25:55 -04:00