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

..

internal

add test scanner (#39419)

2025-07-16 12:45:46 +02:00

main_classes

Use HF papers (#38184)

2025-06-13 11:07:09 +00:00

model_doc

Refactor MambaCache to modeling_mamba.py (#38086)

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

quantization

enable xpu on kv-cache and hqq doc (#39246)

2025-07-07 13:12:02 +00:00

reference

Enhance Model Loading By Providing Parallelism, Uses Optional Env Flag (#36835)

2025-05-23 16:39:47 +00:00

tasks

No more Tuple, List, Dict (#38797)

2025-06-17 19:37:18 +01:00

_config.py

Add optimized PixtralImageProcessorFast (#34836)

2024-11-28 16:04:05 +01:00

_redirects.yml

Docs / Quantization: Redirect deleted page (#31063)

2024-05-28 18:29:22 +02:00

_toctree.yml

GLM-4 Update (#39393)

2025-07-21 13:24:34 +02:00

accelerate.md

change fsdp_strategy to fsdp in TrainingArguments in accelerate doc (#38807)

2025-06-13 15:32:40 +00:00

accelerator_selection.md

[docs] add xpu environment variable for gpu selection (#38194)

2025-05-30 16:05:07 +00:00

add_new_model.md

[docs] Model contribution (#38995)

2025-06-26 12:25:14 -07:00

add_new_pipeline.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

agents.md

[serve] Cursor support, move docs into separate page, add more examples (#39133)

2025-07-03 17:04:16 +01:00

attention_interface.md

🚨🚨 Fix and simplify attention implementation dispatch and subconfigs handling (#39423)

2025-07-18 13:41:54 +02:00

auto_docstring.md

Improve @auto_docstring doc and rename args_doc.py to auto_docstring.py (#39439)

2025-07-18 18:00:34 +00:00

backbones.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

cache_explanation.md

[cache] make all classes cache compatible finally (#38635)

2025-07-16 14:00:17 +02:00

chat_extras.md

Update chat_extras.md with content correction (#36599)

2025-03-07 13:09:02 +00:00

chat_templating_multimodal.md

Enable XPU doc (#38929)

2025-06-30 07:56:55 -07:00

chat_templating_writing.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

chat_templating.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

community.md

Fixed Majority of the Typos in transformers[en] Documentation (#33350)

2024-09-09 10:47:24 +02:00

contributing.md

…

conversations.md

[serve] Cursor support, move docs into separate page, add more examples (#39133)

2025-07-03 17:04:16 +01:00

custom_models.md

No more Tuple, List, Dict (#38797)

2025-06-17 19:37:18 +01:00

debugging.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

deepspeed.md

chore: Fix typos in docs and examples (#36524)

2025-03-04 13:47:41 +00:00

executorch.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

fast_tokenizers.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

feature_extractors.md

Enable XPU doc (#38929)

2025-06-30 07:56:55 -07:00

fsdp.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

generation_features.md

chore: Fix typos in docs and examples (#36524)

2025-03-04 13:47:41 +00:00

generation_strategies.md

Fix custom generate from local directory (#38916)

2025-06-20 17:36:57 +01:00

gguf.md

Fix gguf docs (#36601)

2025-03-11 15:29:14 +01:00

glossary.md

Use HF papers (#38184)

2025-06-13 11:07:09 +00:00

how_to_hack_models.md

[doc] fix bugs in how_to_hack_models.md (#38198)

2025-05-19 10:37:54 -07:00

hpo_train.md

[Nit] Add Note on SigOpt being in Public Archive Mode (#38610)

2025-06-05 14:07:23 -07:00

image_processors.md

🔴 Video processors as a separate class (#35206)

2025-05-12 11:55:51 +02:00

index.md

[docs] Update docs moved to the course (#38800)

2025-06-13 12:02:27 -07:00

installation.md

byebye torch 2.0 (#37277)

2025-04-07 15:19:47 +02:00

kv_cache.md

enable xpu on kv-cache and hqq doc (#39246)

2025-07-07 13:12:02 +00:00

llm_optims.md

[CI] green llama tests (#37244)

2025-04-03 14:15:53 +01:00

llm_tutorial_optimization.md

Use HF papers (#38184)

2025-06-13 11:07:09 +00:00

llm_tutorial.md

No more Tuple, List, Dict (#38797)

2025-06-17 19:37:18 +01:00

model_memory_anatomy.md

Use HF papers (#38184)

2025-06-13 11:07:09 +00:00

model_sharing.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

models.md

Fix grammatical error in models documentation (#39019)

2025-06-25 14:55:22 +00:00

modular_transformers.md

[docs] @auto_docstring (#39011)

2025-06-26 14:21:54 -07:00

notebooks.md

…

optimizers.md

Add StableAdamW Optimizer (#39446)

2025-07-16 13:35:53 +02:00

pad_truncation.md

Fixed Majority of the Typos in transformers[en] Documentation (#33350)

2024-09-09 10:47:24 +02:00

peft.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

perf_hardware.md

Improve grammar and clarity in perf_hardware.md (#39428)

2025-07-16 12:15:15 -07:00

perf_infer_cpu.md

remove ipex_optimize_model usage (#38632)

2025-06-06 20:04:44 +02:00

perf_infer_gpu_multi.md

switch default xpu tp backend to pytorch built-in XCCL from pytorch 2.8 (#39024)

2025-06-30 08:54:05 -07:00

perf_infer_gpu_one.md

Small typo lines 47 and 199 perf_infer_gpu_one.md (#37938)

2025-05-06 14:32:55 +01:00

perf_torch_compile.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

perf_train_cpu_many.md

remove ipex_optimize_model usage (#38632)

2025-06-06 20:04:44 +02:00

perf_train_cpu.md

remove ipex_optimize_model usage (#38632)

2025-06-06 20:04:44 +02:00

perf_train_gaudi.md

Add Intel Gaudi doc (#37855)

2025-04-29 13:28:06 -07:00

perf_train_gpu_many.md

[docs] Tensor parallelism (#38241)

2025-06-26 14:40:45 -07:00

perf_train_gpu_one.md

[docs] Typos - Single GPU efficient training features (#38964)

2025-06-23 12:33:10 -07:00

perf_train_special.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

perplexity.md

[docs] use device-agnostic API instead of cuda (#34913)

2024-11-26 09:23:34 -08:00

philosophy.md

…

pipeline_gradio.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

pipeline_tutorial.md

chore: Fix typos in docs and examples (#36524)

2025-03-04 13:47:41 +00:00

pipeline_webserver.md

fix and enhance pipeline_webserver.md (#36992)

2025-04-15 08:35:05 -07:00

pr_checks.md

Fixed Majority of the Typos in transformers[en] Documentation (#33350)

2024-09-09 10:47:24 +02:00

processors.md

[docs] add Audio import (#38195)

2025-05-19 13:16:35 +00:00

quicktour.md

Add Hugging Face authentication procedure for IDEs (PyCharm, VS Code,… (#38954)

2025-06-24 11:48:15 -07:00

run_scripts.md

Remove research projects (#36645)

2025-03-11 13:47:38 +00:00

serialization.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

serving.md

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

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

testing.md

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

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

tflite.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

tokenizer_summary.md

Use HF papers (#38184)

2025-06-13 11:07:09 +00:00

tools.md

[serve] Cursor support, move docs into separate page, add more examples (#39133)

2025-07-03 17:04:16 +01:00

torchscript.md

Fix wording in torchscript.md (#38004)

2025-05-08 16:47:45 +01:00

trainer.md

[Docs] Fix typo in CustomTrainer compute_loss method and adjust loss reduction logic (#39391)

2025-07-14 09:25:06 -07:00

training.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

troubleshooting.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

video_processors.md

🔴 Video processors as a separate class (#35206)

2025-05-12 11:55:51 +02:00