transformers/tests
Pavel Iakubovskii 048f599f35
Fix RT-DETR weights initialization (#31724)
* Fix init for rt-detr heads

* Fixup

* Add separate prior_prob value to config for initialization

* Add bbox init

* Change to 1 / num_labels init

* Adjust weights init test

* Fix style for test
2024-07-03 14:29:02 +01:00
..
agents Agents: Improve python interpreter (#31409) 2024-06-18 11:55:36 +02:00
benchmark
bettertransformer
deepspeed Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
extended Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
fixtures
fsdp Llama et al. / FSDP : Fix breaking change in 4.40 for FSDP (#31161) 2024-06-26 14:50:08 +01:00
generation fix assisted decoding (#31401) 2024-07-03 09:22:56 +01:00
models Fix RT-DETR weights initialization (#31724) 2024-07-03 14:29:02 +01:00
optimization fix get_scheduler when name is warmup_stable_decay (#31128) 2024-05-30 15:25:43 +01:00
peft_integration
pipelines 🚨🚨 TextGenerationPipeline: rely on the tokenizer default kwargs (#31747) 2024-07-02 16:17:42 +02:00
quantization Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
repo_utils
sagemaker Pass datasets trust_remote_code (#31406) 2024-06-17 17:29:13 +01:00
tokenization Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
trainer add gather_use_object arguments (#31514) 2024-06-28 13:50:27 +01:00
utils Make tool JSON schemas consistent (#31756) 2024-07-02 20:00:42 +01:00
__init__.py
test_backbone_common.py
test_configuration_common.py Move some test files (tets/test_xxx_utils.py) to tests/utils (#31730) 2024-07-02 13:46:03 +02:00
test_feature_extraction_common.py
test_image_processing_common.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
test_image_transforms.py fix: center_crop occasionally outputs off-by-one dimension matrix (#30934) 2024-05-21 13:56:52 +01:00
test_modeling_common.py Move some test files (tets/test_xxx_utils.py) to tests/utils (#31730) 2024-07-02 13:46:03 +02:00
test_modeling_flax_common.py add sdpa to ViT [follow up of #29325] (#30555) 2024-05-16 10:56:11 +01:00
test_modeling_tf_common.py Port IDEFICS to tensorflow (#26870) 2024-05-13 15:59:46 +01:00
test_pipeline_mixin.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
test_processing_common.py add initial design for uniform processors + align model (#31197) 2024-06-13 16:27:16 +02:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00