transformers/tests
Matt c7eb95581a
Don't accidentally mutate the base_model_tp_plan (#36677)
* Don't accidentally mutate the base_model_tp_plan

* Co-authored by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Trigger tests

* Marking grad accum test as slow

* Add a flaky decorator

* Add a flaky decorator

* Use cyril's codeblock

* Don't copy() when it's None

* Use cyril's new codeblock

* make fixup
2025-03-12 18:59:13 +00: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 HPU support (#36424) 2025-03-12 09:08:12 +01:00
extended [tests] skip tests for xpu (#33553) 2024-09-19 19:28:04 +01:00
fixtures Implementation of SuperPoint and AutoModelForKeypointDetection (#28966) 2024-03-19 14:43:02 +00:00
fsdp HPU support (#36424) 2025-03-12 09:08:12 +01:00
generation Don't accidentally mutate the base_model_tp_plan (#36677) 2025-03-12 18:59:13 +00:00
models Gemma3 (#36658) 2025-03-12 09:06:17 +01:00
optimization Support constant lr with cooldown (#35453) 2025-02-10 13:21:55 +01:00
peft_integration Fix pipeline+peft interaction (#36480) 2025-03-03 18:01:43 +00:00
pipelines Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
quantization Fix Expected output for compressed-tensors tests (#36425) 2025-02-26 21:17:24 +01:00
repo_utils [Modular] skip modular checks based on diff (#36130) 2025-02-13 12:53:21 +00:00
sagemaker Trainer - deprecate tokenizer for processing_class (#32385) 2024-10-02 14:08:46 +01:00
tensor_parallel Refactor some core stuff (#36539) 2025-03-11 09:26:28 +01:00
tokenization Fix PretrainedTokenizerFast check => Fix PretrainedTokenizerFast Save (#35835) 2025-02-13 12:00:33 +01:00
trainer Don't accidentally mutate the base_model_tp_plan (#36677) 2025-03-12 18:59:13 +00:00
utils [core] Large/full refactor of from_pretrained (#36033) 2025-03-12 13:39:25 +01:00
__init__.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
test_backbone_common.py Align backbone stage selection with out_indices & out_features (#27606) 2023-12-20 18:33:17 +00:00
test_configuration_common.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
test_feature_extraction_common.py Split common test from core tests (#24284) 2023-06-15 07:30:24 -04:00
test_image_processing_common.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 [core] Large/full refactor of from_pretrained (#36033) 2025-03-12 13:39:25 +01:00
test_modeling_flax_common.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
test_modeling_tf_common.py [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +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 Gemma3 (#36658) 2025-03-12 09:06:17 +01:00
test_sequence_feature_extraction_common.py Fix typo (#25966) 2023-09-05 10:12:25 +02:00
test_tokenization_common.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
test_training_args.py CI: fix test-save-trainer (#36191) 2025-02-14 10:20:56 +01:00