transformers/tests
Yoni Gozlan 93aafdc620
Add compile test for fast image processor (#35184)
* add compile test for fast image processor

* override pixtral test
2024-12-23 13:12:45 -05:00
..
agents Add token cost + runtime monitoring to Agent and HfEngine children (#34548) 2024-12-03 13:14:52 +01:00
benchmark
bettertransformer
deepspeed Trainer - deprecate tokenizer for processing_class (#32385) 2024-10-02 14:08:46 +01:00
extended [tests] skip tests for xpu (#33553) 2024-09-19 19:28:04 +01:00
fixtures
fsdp FSDP grad accum fix (#34645) 2024-11-15 22:28:06 +01:00
generation Make test_generate_with_static_cache even less flaky (#34995) 2024-12-20 16:03:26 +01:00
models Add compile test for fast image processor (#35184) 2024-12-23 13:12:45 -05:00
optimization fix: Fixed the 1st argument name in classmethods (#31907) 2024-07-11 12:11:50 +01:00
peft_integration [PEFT] Better Trainer error when prompt learning with loading best model at the end (#35087) 2024-12-11 12:44:39 +01:00
pipelines Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
quantization HIGGS Quantization Support (#34997) 2024-12-23 16:54:49 +01:00
repo_utils Refactor CI: more explicit (#30674) 2024-08-30 18:17:25 +02:00
sagemaker Trainer - deprecate tokenizer for processing_class (#32385) 2024-10-02 14:08:46 +01:00
tokenization VLM: special multimodal Tokenizer (#34461) 2024-11-04 16:37:51 +01:00
tp Simplify Tensor Parallel implementation with PyTorch TP (#34184) 2024-11-18 19:51:49 +01:00
trainer Fix GA loss bugs and add unit test (#35121) 2024-12-09 09:57:41 +01:00
utils 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
__init__.py
test_backbone_common.py
test_configuration_common.py Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
test_feature_extraction_common.py
test_image_processing_common.py Add compile test for fast image processor (#35184) 2024-12-23 13:12:45 -05:00
test_image_transforms.py
test_modeling_common.py Make test_generate_with_static_cache even less flaky (#34995) 2024-12-20 16:03:26 +01:00
test_modeling_flax_common.py 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
test_modeling_tf_common.py 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
test_pipeline_mixin.py Add image text to text pipeline (#34170) 2024-10-31 15:48:11 -04:00
test_processing_common.py Separate chat templates into a single file (#33957) 2024-11-26 14:18:04 +00:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py Separate chat templates into a single file (#33957) 2024-11-26 14:18:04 +00:00