Yaswanth Gali fbdaa7b099 Add Aimv2 model (#36625) ...

* Model skelton

* changes

* temp push

* changes

* Added support for aimv2-native

* More changes

* More changes

* Stupid mistake correction

* Added config and refactor

* Added vison model

* update

* Refactor for lit variant

* Added Text Model

* Minor fixes

* nits

* update

* Preliminary tests

* More fixes

* Updated tests 🤗

* Refactor

* Updated testcase

* Updated config

* make fixup

* more fixes

* Bug fix and updates

* deadcode

* Fixes

* nit

* up

* Happy CI ✅

* Reduce LOC

* nit

* nit

* make style

* return_dict refactor

* bug fix

* fix

* doc update

* nit

* make fixup

* Minor update

* _init_weigths modifcation

* update tests

* Minor fixes post review

* Update w.r.t GradientCheckpointingLayer

* docs update

* update

* nit

* Use more Modular 😉

* Change name from AIMv2 to Aimv2

* Nit

* make style

* Add model doc pointer

* make style

* Update model doc section

* updates

* Modify attn mask and interface

* update test

* Final change

* Utilize flash and flex attn

* keep attn mask

* camelcase model name in test file

* Fix docstring

* Fix config warning finally and create_causal_mask

* disable torchscript

* remove unused arg

* remove from tests

* balance model size for tests

* fix device

* tests

* tests

* flaky test

* fix import

---------

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

2025-07-08 11:53:21 +02:00

..

internal

Remove all traces of low_cpu_mem_usage (#38792)

2025-06-12 16:39:33 +02:00

main_classes

Use HF papers (#38184)

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

model_doc

Add Aimv2 model (#36625)

2025-07-08 11:53:21 +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

Add Aimv2 model (#36625)

2025-07-08 11:53:21 +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

No more Tuple, List, Dict (#38797)

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

auto_docstring.md

[docs] @auto_docstring (#39011)

2025-06-26 14:21:54 -07: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

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

[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

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

[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

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

Update tiny-agents example (#39245)

2025-07-07 15:58:36 +02: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

feat: add flexible Liger Kernel configuration to TrainingArguments (#38911)

2025-06-19 15:54:08 +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