From 948dd3443bc6b8ffb76cbdddf3f4c5ae0b6637fa Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Tue, 9 Sep 2025 12:40:29 +0800 Subject: [PATCH] [Bugfix] Fix Apertus HF repo name (#24447) Signed-off-by: DarkLight1337 --- docs/models/supported_models.md | 1 + tests/models/language/generation/test_common.py | 2 +- tests/models/registry.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/models/supported_models.md b/docs/models/supported_models.md index bdb29aac333c..d23fdff568fc 100644 --- a/docs/models/supported_models.md +++ b/docs/models/supported_models.md @@ -322,6 +322,7 @@ th { | Architecture | Models | Example HF Models | [LoRA](../features/lora.md) | [PP](../serving/parallelism_scaling.md) | [V1](gh-issue:8779) | |--------------|--------|-------------------|----------------------|---------------------------|---------------------| +| `ApertusForCausalLM` | Apertus | `swiss-ai/Apertus-8B-2509`, `swiss-ai/Apertus-70B-Instruct-2509`, etc. | ✅︎ | ✅︎ | ✅︎ | | `AquilaForCausalLM` | Aquila, Aquila2 | `BAAI/Aquila-7B`, `BAAI/AquilaChat-7B`, etc. | ✅︎ | ✅︎ | ✅︎ | | `ArceeForCausalLM` | Arcee (AFM) | `arcee-ai/AFM-4.5B-Base`, etc. | ✅︎ | ✅︎ | ✅︎ | | `ArcticForCausalLM` | Arctic | `Snowflake/snowflake-arctic-base`, `Snowflake/snowflake-arctic-instruct`, etc. | | ✅︎ | ✅︎ | diff --git a/tests/models/language/generation/test_common.py b/tests/models/language/generation/test_common.py index 8a04946b2ffb..6fc8f1301fdb 100644 --- a/tests/models/language/generation/test_common.py +++ b/tests/models/language/generation/test_common.py @@ -93,7 +93,7 @@ AITER_MODEL_LIST = [ "allenai/OLMoE-1B-7B-0924-Instruct", marks=[pytest.mark.cpu_model], ), - pytest.param("swiss-ai/Apertus-8B"), # apertus + pytest.param("swiss-ai/Apertus-8B-2509"), # apertus ]) @pytest.mark.parametrize("max_tokens", [32]) @pytest.mark.parametrize("num_logprobs", [5]) diff --git a/tests/models/registry.py b/tests/models/registry.py index 8bcdeb087c1f..755a37b109d7 100644 --- a/tests/models/registry.py +++ b/tests/models/registry.py @@ -158,7 +158,7 @@ class _HfExamplesInfo: # yapf: disable _TEXT_GENERATION_EXAMPLE_MODELS = { # [Decoder-only] - "ApertusForCausalLM": _HfExamplesInfo("swiss-ai/Apertus-8B", + "ApertusForCausalLM": _HfExamplesInfo("swiss-ai/Apertus-8B-2509", min_transformers_version="4.56.0", trust_remote_code=True), "AquilaModel": _HfExamplesInfo("BAAI/AquilaChat-7B",