Kyle Sayers 31d81943c9 [Core] [Offloading] Fix saving offloaded submodules (#39280) ...

* fix counting meta tensors, fix onloading meta tensors

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

* remove unrelated fix

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

* remove unrelated change

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

* add clarifying comment

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

* add test_save_offloaded_model_with_direct_params

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

* fix merge conflict, add decorators

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

---------

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

2025-07-16 08:44:40 +00:00

..

import_structures

Support for version spec in requires & arbitrary mismatching depths across folders (#37854)

2025-05-09 15:26:27 +02:00

__init__.py

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

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

test_activations.py

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

test_add_new_model_like.py

[tests] remove tests from libraries with deprecated support (flax, tensorflow_text, ...) (#39051)

2025-06-26 16:25:00 +01:00

test_audio_utils.py

Add torchcodec in docstrings/tests for datasets 4.0 (#39156)

2025-07-08 17:06:12 +02:00

test_auto_docstring.py

[AutoDocstring] Based on inspect parsing of the signature (#33771)

2025-05-08 17:46:07 -04:00

test_backbone_utils.py

🚨 out_indices always a list (#30941)

2024-05-22 15:23:04 +01:00

test_cache_utils.py

fix(generation): stop beam search per-instance when heuristic satisfied (#38778)

2025-07-08 08:59:37 +00:00

test_chat_template_utils.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_cli.py

Transformers cli clean command (#37657)

2025-04-30 12:15:43 +01:00

test_configuration_utils.py

🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)

2025-05-23 17:17:38 +02: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_deprecation.py

enable utils test cases on XPU (#38005)

2025-05-09 08:45:01 +02:00

test_doc_samples.py

[tests] remove TF tests (uses of require_tf) (#38944)

2025-06-25 17:29:10 +00: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_expectations.py

Expectation fixes and added AMD expectations (#38729)

2025-06-13 16:14:58 +02:00

test_feature_extraction_utils.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_file_utils.py

[tests] remove tests from libraries with deprecated support (flax, tensorflow_text, ...) (#39051)

2025-06-26 16:25:00 +01:00

test_generic.py

Refactor the way we handle outputs for new llamas and new models (#39120)

2025-07-05 11:34:28 +02:00

test_hf_argparser.py

No more Tuple, List, Dict (#38797)

2025-06-17 19:37:18 +01:00

test_hub_utils.py

Add test to ensure unknown exceptions reraising in utils/hub.py::cached_files() (#37651)

2025-04-22 11:38:10 +02:00

test_image_processing_utils.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_image_utils.py

🔴 Video processors as a separate class (#35206)

2025-05-12 11:55:51 +02:00

test_import_structure.py

Support for version spec in requires & arbitrary mismatching depths across folders (#37854)

2025-05-09 15:26:27 +02:00

test_import_utils.py

Fix test isolation for clear_import_cache utility (#36345)

2025-03-17 16:09:09 +01:00

test_logging.py

Fix flaky test for log level (#21776)

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

test_masking_utils.py

Add a default value for position_ids in masking_utils (#39310)

2025-07-10 18:53:40 +02:00

test_model_card.py

Use HF papers (#38184)

2025-06-13 11:07:09 +00:00

test_model_debugging_utils.py

Model debugger upgrades (#37391)

2025-04-18 16:45:54 +02:00

test_model_output.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_modeling_rope_utils.py

Delete deprecated stuff (#38838)

2025-07-10 05:18:44 +00:00

test_modeling_utils.py

[Core] [Offloading] Fix saving offloaded submodules (#39280)

2025-07-16 08:44:40 +00:00

test_offline.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_skip_decorators.py

enable utils test cases on XPU (#38005)

2025-05-09 08:45:01 +02:00

test_tokenization_utils.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_versions_utils.py

Remove old code for PyTorch, Accelerator and tokenizers (#37234)

2025-04-10 20:54:21 +02:00

test_video_utils.py

[video processor] support torchcodec and decrease cuda memory usage (#38880)

2025-06-25 08:23:37 +00:00

tiny_model_summary.json

Add image text to text pipeline (#34170)

2024-10-31 15:48:11 -04:00