transformers/tests
Yao Matrix 8f6b27eb5c
enable test_assisted_decoding_in_different_gpu test on XPU (#37120)
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
2025-04-01 11:22:59 +02:00
..
agents
bettertransformer
deepspeed chore: fix typos in the tests directory (#36813) 2025-03-21 10:20:05 +01:00
extended
fixtures
fsdp
generation enable test_assisted_decoding_in_different_gpu test on XPU (#37120) 2025-04-01 11:22:59 +02:00
models Fix llava xpu tests. (#37130) 2025-04-01 11:10:13 +02:00
optimization
peft_integration Set weights_only in torch.load (#36991) 2025-03-27 14:55:50 +00:00
pipelines Update ruff to 0.11.2 (#36962) 2025-03-25 16:00:11 +01:00
quantization add gpt2 test on XPU (#37028) 2025-04-01 11:09:29 +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 Fix more inefficient PT operations (#37060) 2025-03-31 16:31:24 +01:00
__init__.py
test_backbone_common.py
test_configuration_common.py
test_feature_extraction_common.py
test_image_processing_common.py
test_image_transforms.py
test_modeling_common.py Fix more inefficient PT operations (#37060) 2025-03-31 16:31:24 +01:00
test_modeling_flax_common.py
test_modeling_tf_common.py
test_pipeline_mixin.py
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