transformers/tests
Xavier Dupré 6f5dc9c82e
Fixes DynamicCache export issues due to control flow and inplace modifications (#36652)
* Remove unnecessary masked_fill in deberta models

* Enable some code when exporting but not compiling

* add missing import

* style

* replace if by torch.cond

* style

* use numel

* style

* add unit tests

* style

* change empty value for dynamic cache

* replace != [] by numel()

* fix import issue

* style
2025-04-02 12:04:40 +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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
deepspeed chore: fix typos in the tests directory (#36813) 2025-03-21 10:20:05 +01:00
extended [tests] skip tests for xpu (#33553) 2024-09-19 19:28:04 +01:00
fixtures
fsdp HPU support (#36424) 2025-03-12 09:08:12 +01:00
generation Fixes DynamicCache export issues due to control flow and inplace modifications (#36652) 2025-04-02 12:04:40 +01:00
models [chat-template] fix video loading (#37146) 2025-04-02 11:27:50 +02:00
optimization Just import torch AdamW instead (#36177) 2025-03-19 18:29:40 +00:00
peft_integration Set weights_only in torch.load (#36991) 2025-03-27 14:55:50 +00:00
pipelines Avoid pipeline test failing related to Hub call (#37170) 2025-04-01 18:22:45 +02:00
quantization Add device workaround for int4 weight only quantization after API update (#36980) 2025-04-02 12:42:22 +02:00
repo_utils Adding Qwen3 and Qwen3MoE (#36878) 2025-03-31 09:50:49 +02:00
sagemaker Change GPUS to GPUs (#36945) 2025-03-25 17:25:39 +01:00
tensor_parallel enable tp on CPU (#36299) 2025-03-31 10:55:47 +02:00
tokenization Use lru_cache for tokenization tests (#36818) 2025-03-28 15:09:35 +01:00
trainer Set weights_only in torch.load (#36991) 2025-03-27 14:55:50 +00:00
utils Convert _VALID_DICT_FIELDS to class attribute for shared dict parsing in subclasses (#36736) 2025-04-01 12:29:12 +02:00
__init__.py
test_backbone_common.py
test_configuration_common.py Refactor Attention implementation for ViT-based models (#36545) 2025-03-20 15:15:01 +00:00
test_feature_extraction_common.py
test_image_processing_common.py Make the flaky list a little more general (#36704) 2025-03-14 12:15:32 +00:00
test_image_transforms.py Uses Collection in transformers.image_transforms.normalize (#36301) 2025-02-21 18:38:41 +01:00
test_modeling_common.py Refactor attention for SigLIP based models (#36981) 2025-04-01 15:37:25 +02:00
test_modeling_flax_common.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
test_modeling_tf_common.py fix typos in the tests directory (#36717) 2025-03-17 17:45:57 +00:00
test_pipeline_mixin.py Add image text to text pipeline (#34170) 2024-10-31 15:48:11 -04:00
test_processing_common.py [chat templates} support loading audio from video (#36955) 2025-03-27 14:46:11 +01:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py Use lru_cache for tokenization tests (#36818) 2025-03-28 15:09:35 +01:00
test_training_args.py Fix TrainingArguments.torch_empty_cache_steps post_init check (#36734) 2025-03-17 16:09:46 +01:00