[Model] Remove AquilaForCausalLM, AquilaModel (#46605)
Signed-off-by: Xianbao QIAN <xianbao.qian@gmail.com> Co-authored-by: Claude <noreply@anthropic.com>
Tiezhen WANG committed
Jun 25, 2026 at 16:08 UTC
72adb20a6ac023fa6a6b2748fda673e582d2f74e
4 files changed
+2
-7
docs/models/supported_models.md
-1
@@ -366,7 +366,6 @@ th {
366
| ------------ | ------ | ----------------- | -------------------- | ------------------------- |
367
| `AfmoeForCausalLM` | Afmoe | TBA | ✅︎ | ✅︎ |
368
| `ApertusForCausalLM` | Apertus | `swiss-ai/Apertus-8B-2509`, `swiss-ai/Apertus-70B-Instruct-2509`, etc. | ✅︎ | ✅︎ |
369
-| `AquilaForCausalLM` | Aquila, Aquila2 | `BAAI/Aquila-7B`, `BAAI/AquilaChat-7B`, etc. | ✅︎ | ✅︎ |
369
| `ArceeForCausalLM` | Arcee (AFM) | `arcee-ai/AFM-4.5B-Base`, etc. | ✅︎ | ✅︎ |
370
| `ArcticForCausalLM` | Arctic | `Snowflake/snowflake-arctic-base`, `Snowflake/snowflake-arctic-instruct`, etc. | | ✅︎ |
371
| `AXK1ForCausalLM` | A.X-K1 | `skt/A.X-K1`, etc. | | ✅︎ |
tests/distributed/test_pipeline_parallel.py
-2
@@ -102,8 +102,6 @@ class PPTestSettings:
102
103
TEXT_GENERATION_MODELS = {
104
# [Decoder-only]
105
- # Uses Llama
106
- # "BAAI/AquilaChat-7B": PPTestSettings.fast(),
105
"Snowflake/snowflake-arctic-instruct": PPTestSettings.fast(load_format="dummy"),
106
"bigscience/bloomz-1b1": PPTestSettings.fast(),
107
"zai-org/chatglm3-6b": PPTestSettings.fast(),
tests/models/registry.py
-2
@@ -201,8 +201,6 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
201
# [Decoder-only]
202
"AfmoeForCausalLM": _HfExamplesInfo("arcee-ai/Trinity-Nano-Preview"),
203
"ApertusForCausalLM": _HfExamplesInfo("swiss-ai/Apertus-8B-Instruct-2509"),
204
- "AquilaModel": _HfExamplesInfo("BAAI/AquilaChat-7B", trust_remote_code=True),
205
- "AquilaForCausalLM": _HfExamplesInfo("BAAI/AquilaChat2-7B", trust_remote_code=True),
204
"ArceeForCausalLM": _HfExamplesInfo("arcee-ai/AFM-4.5B-Base"),
205
"ArcticForCausalLM": _HfExamplesInfo(
206
"Snowflake/snowflake-arctic-instruct", trust_remote_code=True
vllm/model_executor/models/registry.py
+2
-2
@@ -72,8 +72,6 @@ _TEXT_GENERATION_MODELS = {
72
# [Decoder-only]
73
"AfmoeForCausalLM": ("afmoe", "AfmoeForCausalLM"),
74
"ApertusForCausalLM": ("apertus", "ApertusForCausalLM"),
75
- "AquilaModel": ("llama", "LlamaForCausalLM"),
76
- "AquilaForCausalLM": ("llama", "LlamaForCausalLM"), # AquilaChat2
75
"ArceeForCausalLM": ("arcee", "ArceeForCausalLM"),
76
"ArcticForCausalLM": ("arctic", "ArcticForCausalLM"),
77
"AXK1ForCausalLM": ("AXK1", "AXK1ForCausalLM"),
@@ -731,6 +729,8 @@ _PREVIOUSLY_SUPPORTED_MODELS = {
729
"MiniMaxVL01ForConditionalGeneration": "0.23.0",
730
"BaiChuanForCausalLM": "0.23.0",
731
"BaichuanForCausalLM": "0.23.0",
732
+ "AquilaModel": "0.24.0",
733
+ "AquilaForCausalLM": "0.24.0",
734
}
735
736
_OOT_SUPPORTED_MODELS = {