Manuel de Prada Corral 1aa7256f01 Refactor MambaCache to modeling_mamba.py (#38086) ...

* Refactor MambaCache to modeling_mamba.py (parity with Zamba)

* ruff

* fix dummies

* update

* update

* remove mamba ref in cache tests

* remove cache_implementation from tests

* update

* ruff

* ruff

* sneaky regression

* model consistency

* fix test_multi_gpu_data_parallel_forward

* fix falcon slow tests

* ruff

* ruff

* add sample false

* try to fix slow tests

* Revert "fix test_multi_gpu_data_parallel_forward"

This reverts commit 66b7162c7c5c5ce8a73ccf48cffc8a96343ebb33.

* fix tests on nvidia t4, remove dataparallel tests from mamba

* ruff

* remove DDP tests from mamba and falcon_mamba

* add explicit error for MambaCache

* mamba2 also needs to init cache in prepare_inputs_for_generation

* ruff

* ruff

* move MambaCache to its own file

* ruff

* unprotected import fix

* another attempt to fix unprotected imports

* Revert "another attempt to fix unprotected imports"

This reverts commit 2338354fcab630de5899321f5daced5fb312c2a2.

* fixing unprotected import, attempt 3

* Update src/transformers/cache_utils.py

* ruff's fault

* fix arthur review

* modular falcon mamba

* found a hack

* fix config docs

* fix docs

* add export info

* merge modular falcon branch

* oopsie

* fix fast path failing

* new approach

* oopsie

* fix types

* Revert new pragma in modular

This reverts commit 80b1cf160ee251536f07c40b8a0857d499e70db6.

* trying another modular workaround

* review & fix ci

* oopsie

* clear prepare_inputs on mamba/mamba2/falcon_mamba

2025-07-21 14:59:36 +02:00

..

bettertransformer

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

commands

[serve] Add speech to text (/v1/audio/transcriptions) (#39434)

2025-07-17 14:29:57 +00:00

deepspeed

Remove script datasets in tests (#38940)

2025-06-25 14:31:20 +00:00

extended

Add Optional to remaining types (#37808)

2025-04-28 14:20:45 +01:00

fixtures

[tests] remove tests from libraries with deprecated support (flax, tensorflow_text, ...) (#39051)

2025-06-26 16:25:00 +01:00

fsdp

Gaudi3 CI (#38790)

2025-06-23 10:56:51 +02:00

generation

Rename _supports_flash_attn_2 in examples and tests (#39471)

2025-07-21 14:02:57 +02:00

models

Refactor MambaCache to modeling_mamba.py (#38086)

2025-07-21 14:59:36 +02:00

optimization

[tests] remove TF tests (uses of require_tf) (#38944)

2025-06-25 17:29:10 +00:00

peft_integration

Enable some ruff checks for performance and readability (#39383)

2025-07-17 13:21:59 +00:00

pipelines

Raise TypeError instead of ValueError for invalid types (#38660)

2025-07-21 12:42:00 +00:00

quantization

Remove residual quantization attribute from dequantized models (#39373)

2025-07-15 17:16:10 +02:00

repo_utils

[modular] speedup check_modular_conversion with multiprocessing (#37456)

2025-07-10 19:07:59 +01:00

sagemaker

Deprecate TF + JAX (#38758)

2025-06-11 17:28:06 +01:00

tensor_parallel

enable static cache on TP model (#39164)

2025-07-09 21:14:45 +00:00

tokenization

[tests] remove tests from libraries with deprecated support (flax, tensorflow_text, ...) (#39051)

2025-06-26 16:25:00 +01:00

trainer

Fix tests due to breaking change in accelerate (#39451)

2025-07-17 13:51:50 +01:00

utils

Refactor MambaCache to modeling_mamba.py (#38086)

2025-07-21 14:59:36 +02:00

__init__.py

…

causal_lm_tester.py

Rename _supports_flash_attn_2 in examples and tests (#39471)

2025-07-21 14:02:57 +02:00

test_backbone_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_configuration_common.py

Update composition flag usage (#36263)

2025-04-09 11:48:49 +02:00

test_feature_extraction_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_image_processing_common.py

Fix overriding Fast Image/Video Processors instance attributes affect other instances (#39363)

2025-07-12 23:39:06 +00:00

test_image_transforms.py

Raise TypeError instead of ValueError for invalid types (#38660)

2025-07-21 12:42:00 +00:00

test_modeling_common.py

Fix missing initializations for models created in 2023 (#39239)

2025-07-21 14:43:52 +02:00

test_pipeline_mixin.py

Enable some ruff checks for performance and readability (#39383)

2025-07-17 13:21:59 +00:00

test_processing_common.py

[chat template] return assistant mask in processors (#38545)

2025-07-18 12:23:20 +00:00

test_sequence_feature_extraction_common.py

[tests] remove TF tests (uses of require_tf) (#38944)

2025-06-25 17:29:10 +00:00

test_tokenization_common.py

Fix pylint warnings (#39477)

2025-07-21 12:38:05 +00:00

test_tokenization_mistral_common.py

Add voxtral (#39429)

2025-07-18 00:02:04 +00:00

test_training_args.py

Fix TrainingArguments.torch_empty_cache_steps post_init check (#36734)

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

test_video_processing_common.py

Fix overriding Fast Image/Video Processors instance attributes affect other instances (#39363)

2025-07-12 23:39:06 +00:00