transformers/tests
Yih-Dar c600e89f5c
Update unwrap_and_save_reload_schedule to use weights_only=False (#35952)
* fix

* Fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-01-29 14:30:57 +01:00
..
agents use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
bettertransformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
deepspeed use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
extended
fixtures
fsdp
generation fix test_generated_length_assisted_generation (#34935) 2025-01-29 12:03:45 +00:00
models Test: generate with torch.compile(model.forward) as a fast test (#34544) 2025-01-28 14:10:38 +00:00
optimization Update unwrap_and_save_reload_schedule to use weights_only=False (#35952) 2025-01-29 14:30:57 +01:00
peft_integration use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
pipelines Fix test_pipelines_video_classification that was always failing (#35842) 2025-01-23 19:22:32 +01:00
quantization Split and clean up GGUF quantization tests (#35502) 2025-01-27 15:46:57 +01:00
repo_utils
sagemaker
tokenization
tp Fix TP initialization (#35860) 2025-01-28 15:07:37 +01:00
trainer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
utils Test: generate with torch.compile(model.forward) as a fast test (#34544) 2025-01-28 14:10:38 +00:00
__init__.py
test_backbone_common.py
test_configuration_common.py
test_feature_extraction_common.py
test_image_processing_common.py use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
test_image_transforms.py
test_modeling_common.py Fix: loading DBRX back from saved path (#35728) 2025-01-28 11:38:45 +01:00
test_modeling_flax_common.py
test_modeling_tf_common.py
test_pipeline_mixin.py
test_processing_common.py
test_sequence_feature_extraction_common.py
test_tokenization_common.py