Armaghan Shakir 55736eea99 Add support for MiniMax's MiniMax-Text-01 (#35831) ...

* end-to-end architecture

* lightning-attn: refactor, clean, optimize

* put minimax_text_01 in other files

* use latest __init__ standards and auto-generate modular

* support attention_mask for lightning-attn

* Revert "use latest __init__ standards and auto-generate modular"

This reverts commit d8d3c409d89e335c98a8cd36f47304a76eac7493.

* fix modular conversion

* pass both attention masks instead of tuple

* formatting

* Updated Dynamic Cache

* created MiniMaxText01Cache

* fix hardcoded slope_rate

* update attn_type_list in config

* fix lightning when use_cache=False

* copy tests from mixtral

* (checkpoint) all tests pass for normal attention

* fix all unittests

* fix import sorting

* fix consistency and formatting tests

* fix config

* update tests, since changes in main

* fix seq_len error

* create dummy docs

* fix checkpoint

* add checkpoint in config docstring

* run modular_conversion

* update docs

* fix checkpoint path and update tests

* fix ruff

* remove repeated expected_slice

* update docs

* rename "minimax-text-01" to "minimax"

* inherit config from mixtral

* remove from docs in other languages

* undo files that should be untouched

* move minimax to end in conversation docs

* use MiniMaxForCausalLM as it is

* ruff fixes

* run modular

* fix docstring example in causallm

* refactor attention loop and decay factors

* refactor config in modular

* run modular

* refactor cache

* rename static_cache to linear_cache

* make positional embeddings necessary

* remove unnecessary layernorms declarations

* fix import in tests

* refactor attention in next tokens

* remove outdated code

* formatting and modular

* update tests

* rename layernorm alpha/beta factors

* register decay factors as buffers

* remove unused declarations of decay factors

* update config for alpha/beta factors

* run modular

* remove head_dim in tests

* remove minimax from fx.py

* remove stuff that is not really needed

* update __init__

* update qkv torch.split

Co-authored-by: Cyril Vallez <cyril.vallez@gmail.com>

* fix qkv torch.split

* quality fixes

* remove mistakenly added dummy

* purge unused ModelTester code

* fix-copies

* run fix-copies

* fix head_dim

* write cache formatting tests

* remove postnorm

* avoid contiguous in attention current states

* update expected_slice

* add generation test for integration

* fix dtype in generation test

* update authors

* update with changes in main

* update graident checkpointing and minor fixes

* fix mutable attn_type_list

* rename: attn_type -> layer_type

* update for layer_types

* update integration tests

* update checkpoint

* clean overview in docs

---------

Co-authored-by: Shakib-IO <shakib.khan17@northsouth.edu>
Co-authored-by: Cyril Vallez <cyril.vallez@gmail.com>

2025-06-04 09:38:40 +02:00

..

internal

[generate] move SinkCache to a custom_generate repo (#38399)

2025-06-02 12:13:30 +02:00

main_classes

fix typos (#38336)

2025-05-26 14:42:37 +00:00

model_doc

Add support for MiniMax's MiniMax-Text-01 (#35831)

2025-06-04 09:38:40 +02:00

quantization

Name change AOPermod -> ModuleFqn (#38456)

2025-06-03 15:43:31 +00:00

reference

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

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

tasks

Enhance documentation to explain chat-based few-shot prompting (#37828)

2025-04-30 11:00:10 -07: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

Add support for MiniMax's MiniMax-Text-01 (#35831)

2025-06-04 09:38:40 +02:00

accelerate.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08: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

Transformers cli clean command (#37657)

2025-04-30 12:15:43 +01:00

add_new_pipeline.md

[docs] Redesign (#31757)

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

agents.md

[agents] remove agents 🧹 (#37368)

2025-04-11 18:42:37 +01:00

attention_interface.md

🚨🚨[core] Completely rewrite the masking logic for all attentions (#37866)

2025-05-22 11:38:26 +02:00

attention.md

[Docs] Fix broken links and syntax issues (#28918)

2024-02-08 14:13:35 -08:00

auto_docstring.md

[AutoDocstring] Based on inspect parsing of the signature (#33771)

2025-05-08 17:46:07 -04:00

backbones.md

[docs] Redesign (#31757)

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

cache_explanation.md

[docs] Format fix (#38414)

2025-06-03 09:53:23 -07:00

chat_extras.md

Update chat_extras.md with content correction (#36599)

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

chat_templating_multimodal.md

[chat-template] Unify tests and clean up 🧼 (#37275)

2025-04-10 14:42:32 +02: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

[chat] generate parameterization powered by GenerationConfig and UX-related changes (#38047)

2025-05-12 14:04:41 +01:00

custom_models.md

Fix typos (#36910)

2025-03-24 14:08:29 +00: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

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08: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

[custom_generate] don't forward custom_generate and trust_remote_code (#38304)

2025-05-23 14:49:39 +00:00

gguf.md

Fix gguf docs (#36601)

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

glossary.md

Fix typos (#31819)

2024-07-08 11:52:47 +01: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

[docs] Redesign (#31757)

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

image_processors.md

🔴 Video processors as a separate class (#35206)

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

index.md

Adding Qwen3 and Qwen3MoE (#36878)

2025-03-31 09:50:49 +02:00

installation.md

byebye torch 2.0 (#37277)

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

kv_cache.md

[generate] move SinkCache to a custom_generate repo (#38399)

2025-06-02 12:13:30 +02:00

llm_optims.md

[CI] green llama tests (#37244)

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

llm_tutorial_optimization.md

fix typos in the docs directory (#36639)

2025-03-11 09:41:41 -07:00

llm_tutorial.md

[custom_generate] don't forward custom_generate and trust_remote_code (#38304)

2025-05-23 14:49:39 +00:00

model_memory_anatomy.md

Enable BNB multi-backend support (#31098)

2024-09-24 03:40:56 -06:00

model_sharing.md

[docs] Redesign (#31757)

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

model_summary.md

model_summary.md - Restore link to Harvard's Annotated Transformer. (#29702)

2024-03-23 18:29:39 -07:00

models.md

[docs] minor fixes in models.md (#38193)

2025-05-19 13:14:21 +00:00

modular_transformers.md

Never fallback to eager implicitly (#38327)

2025-05-23 19:48:01 +02:00

notebooks.md

…

optimizers.md

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08: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

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

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

perf_infer_cpu.md

[docs] Redesign (#31757)

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

perf_infer_gpu_multi.md

Fix: make docs work better with doc builder (#38213)

2025-05-20 08:23:03 +00: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

[docs] Redesign (#31757)

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

perf_train_cpu.md

[docs] Redesign (#31757)

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

perf_train_gaudi.md

Add Intel Gaudi doc (#37855)

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

perf_train_gpu_many.md

docs: fix typo (#37567)

2025-04-17 14:54:44 +01:00

perf_train_gpu_one.md

[docs] Redesign (#31757)

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

perf_train_special.md

[docs] Redesign (#31757)

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

perf_train_tpu_tf.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

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

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

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08: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

fix docs serving typos. (#37936)

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

task_summary.md

[doctest] Fixes (#35863)

2025-01-26 15:26:38 -08:00

tasks_explained.md

fix: Wrong task mentioned in docs (#34757)

2024-11-18 18:42:28 +00:00

testing.md

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

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

tf_xla.md

[docs] Redesign (#31757)

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

tflite.md

[docs] Redesign (#31757)

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

tokenizer_summary.md

[docs] Spanish translation of tokenizer_summary.md (#31154)

2024-06-03 16:52:23 -07:00

tools.md

[agents] remove agents 🧹 (#37368)

2025-04-11 18:42:37 +01:00

torchscript.md

Fix wording in torchscript.md (#38004)

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

trainer.md

Update trainer.md (#38113)

2025-05-14 12:40:00 +00: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