Matthew Hoffman d4bd33cc9f Register ModelOutput subclasses as supported torch.utils._pytree nodes (#25358) ...

* Register ModelOutput subclasses as supported torch.utils._pytree nodes

Fixes #25357 where DDP with static_graph=True does not sync gradients when calling backward() over tensors contained in ModelOutput subclasses

* Add test for torch pytree ModelOutput serialization and deserialization

2023-08-08 08:12:11 +02:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_activations_tf.py

TF: Add sigmoid activation function (#16819)

2022-04-19 16:13:08 +01:00

test_activations.py

Add the GeLU activation from pytorch with the tanh approximation (#21345)

2023-02-02 09:33:04 -05:00

test_add_new_model_like.py

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

test_audio_utils.py

audio_utils improvements (#21998)

2023-05-09 09:10:17 -04:00

test_backbone_utils.py

Add methods to update and verify out_features out_indices (#23031)

2023-05-04 10:15:06 +01:00

test_cli.py

add trust_remote_code option to CLI download cmd (#24097)

2023-06-08 11:13:57 -04:00

test_convert_slow_tokenizer.py

Revert error back into warning for byte fallback conversion. (#22607)

2023-04-06 14:00:29 +02:00

test_doc_samples.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_dynamic_module_utils.py

Fix the regex in get_imports to support multiline try blocks and excepts with specific exception types (#23725)

2023-05-24 15:40:19 -04:00

test_file_utils.py

Inheritance-based framework detection (#21784)

2023-02-27 15:31:55 +00:00

test_generic.py

Refactor conversion function (#19799)

2022-10-24 13:48:40 -04:00

test_hf_argparser.py

⚠️ Time to say goodbye to py37 (#24091)

2023-06-28 07:22:39 +02:00

test_hub_utils.py

Add test when downloading from gated repo (#25039)

2023-07-28 08:14:27 -04:00

test_image_processing_utils.py

Add Image Processors (#19796)

2022-11-02 11:57:36 +00:00

test_image_utils.py

Add timeout parameter to load_image function (#25184)

2023-08-03 15:51:54 +01:00

test_logging.py

Fix flaky test for log level (#21776)

2023-02-28 16:24:14 -05:00

test_model_card.py

Black preview (#17217)

2022-05-12 16:25:55 -04:00

test_model_output.py

Register ModelOutput subclasses as supported torch.utils._pytree nodes (#25358)

2023-08-08 08:12:11 +02:00

test_modeling_tf_core.py

Make (TF) CI faster (test only a subset of model classes) (#24592)

2023-06-30 16:54:54 +02:00

test_offline.py

Fix the expected error in test_offline_mode_pipeline_exception (#23022)

2023-04-27 14:22:05 +02:00

test_skip_decorators.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_versions_utils.py

⚠️ Time to say goodbye to py37 (#24091)

2023-06-28 07:22:39 +02:00

tiny_model_summary.json

Update tiny model info. and pipeline testing (#25213)

2023-07-31 19:35:33 +02:00