Pavel Iakubovskii a1e389e637 Refactor return_dict logic to remove complicated if/else paths (#36794) ...

* SAM

* CLIP

* SigLIP

* GOT-OCR2 (depends on SAM)

* SigLIP2 (depends on SigLIP)

* trigger tests

* Fix SAM

* Fix missed indexing, use named attributes

* Llama

* Aria

* Bamba

* Update llama: missed outputs return type

* (fixup) Aria

* DiffLlama

* Emu3

* Gemma

* Gemma2

* Paligemma

* Fix paligemma

* Gemma3

* GLM

* Helium

* JetMoe

* Jamba

* Mistral

* Mistral

* Mixtral

* Nemotron

* Olmo

* Olmo2

* Persimmon

* Phi

* Phi3

* PhiMoe

* Qwen2

* Qwen2_moe

* StableLM

* Starcoder2

* Add return_dict decorator

* SAM

* Update decorator: compile, export, trace - friendly

* Llama (decorator)

* SAM (decorator)

* Add decorator `can_return_tuple`

* Llama

* Update to decorator

* Update CLIP

* Update decorator to store `_is_top_level_module` in self

* Update decorator to correctly handle compile/export

* Remove is_torchdynamo_compiling constraint, all work fine with self attribute assignment

* Typing

* GPT NeoX

* Fixup

* Fix attribute Granite

* Fix return type mixtral

* Update Gemma3

* Fix Cohere amd Cohere2

* Fixup

* Fix corner case for Phi4, when activation is shared

* (fix-copies) deepseekv3, phi4

* Fixup

* Apply to qwen3/qwen3_moe

* Fix

2025-03-31 16:23:37 +01:00

..

import_structures

Import structure & first three model refactors (#31329)

2024-09-10 11:10:53 +02:00

__init__.py

…

test_activations_tf.py

TF: Add sigmoid activation function (#16819)

2022-04-19 16:13:08 +01:00

test_activations.py

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

test_add_new_model_like.py

fix typos in the tests directory (#36717)

2025-03-17 17:45:57 +00:00

test_audio_utils.py

[audio utils] fix fft_bin_width computation (#36603)

2025-03-27 15:20:02 +01:00

test_backbone_utils.py

🚨 out_indices always a list (#30941)

2024-05-22 15:23:04 +01:00

test_cache_utils.py

Remove deprecated batch_size parameter (#37007)

2025-03-27 15:01:56 +00:00

test_chat_template_utils.py

Tool calling: support more types (#35776)

2025-01-20 19:15:34 +01:00

test_cli.py

Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)

2024-08-23 11:12:53 +01:00

test_configuration_utils.py

Fix flaky Hub CI (test_trainer.py) (#35062)

2024-12-05 17:02:27 +01:00

test_convert_slow_tokenizer.py

Revert error back into warning for byte fallback conversion. (#22607)

2023-04-06 14:00:29 +02:00

test_deprecation.py

[Backend support] Allow num_logits_to_keep as Tensor + add flag (#35757)

2025-01-23 09:47:54 +01:00

test_doc_samples.py

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

test_dynamic_module_utils.py

Fix the regex in get_imports to support multiline try blocks and excepts with specific exception types (#23725)

2023-05-24 15:40:19 -04:00

test_expectations.py

Expectations test utils (#36569)

2025-03-18 23:39:50 +01:00

test_feature_extraction_utils.py

Fix flaky Hub CI (test_trainer.py) (#35062)

2024-12-05 17:02:27 +01:00

test_file_utils.py

Inheritance-based framework detection (#21784)

2023-02-27 15:31:55 +00:00

test_generic.py

Refactor return_dict logic to remove complicated if/else paths (#36794)

2025-03-31 16:23:37 +01:00

test_hf_argparser.py

HfArgumentParser: allow for hyhenated field names in long-options (#33990)

2024-10-10 11:58:26 +02:00

test_hub_utils.py

[core] Large/full refactor of from_pretrained (#36033)

2025-03-12 13:39:25 +01:00

test_image_processing_utils.py

Fix flaky Hub CI (test_trainer.py) (#35062)

2024-12-05 17:02:27 +01:00

test_image_utils.py

Fix typos (#36910)

2025-03-24 14:08:29 +00:00

test_import_structure.py

Fix some missing tests in circleci (#33559)

2024-09-20 20:58:51 +02:00

test_import_utils.py

Fix test isolation for clear_import_cache utility (#36345)

2025-03-17 16:09:09 +01:00

test_logging.py

Fix flaky test for log level (#21776)

2023-02-28 16:24:14 -05:00

test_model_card.py

Automatically add transformers tag to the modelcard (#32623)

2024-08-13 07:59:01 +02:00

test_model_output.py

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

test_modeling_flax_utils.py

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

test_modeling_rope_utils.py

fix typos in the tests directory (#36717)

2025-03-17 17:45:57 +00:00

test_modeling_tf_core.py

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

test_modeling_tf_utils.py

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

test_modeling_utils.py

[Feature] Support using FlashAttention2 on Ascend NPU (#36696)

2025-03-31 16:12:58 +02:00

test_offline.py

Final CI cleanup (#36703)

2025-03-13 17:26:09 +00:00

test_processing_utils.py

Uniformize kwargs for Pixtral processor (#33521)

2024-09-17 14:44:27 -04:00

test_skip_decorators.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_tokenization_utils.py

Fix flaky Hub CI (test_trainer.py) (#35062)

2024-12-05 17:02:27 +01:00

test_versions_utils.py

improve dev setup comments and hints (#28495)

2024-01-15 18:36:40 +00:00

tiny_model_summary.json

Add image text to text pipeline (#34170)

2024-10-31 15:48:11 -04:00