From 82e7f9bb031a9aabb320db88503b6f948ded6e95 Mon Sep 17 00:00:00 2001 From: "rongfu.leng" Date: Wed, 14 May 2025 17:13:47 +0800 Subject: [PATCH] [Misc] replace does not exist model (#18119) Signed-off-by: rongfu.leng --- docs/source/models/supported_models.md | 2 +- tests/distributed/test_pipeline_parallel.py | 2 +- tests/models/registry.py | 2 +- vllm/test_utils.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/source/models/supported_models.md b/docs/source/models/supported_models.md index 4bb831749287f..80eccfd034af2 100644 --- a/docs/source/models/supported_models.md +++ b/docs/source/models/supported_models.md @@ -539,7 +539,7 @@ Specified using `--task generate`. * ✅︎ - * `OLMo2ForCausalLM` * OLMo2 - * `allenai/OLMo2-7B-1124`, etc. + * `allenai/OLMo-2-0425-1B`, etc. * * ✅︎ - * `OLMoEForCausalLM` diff --git a/tests/distributed/test_pipeline_parallel.py b/tests/distributed/test_pipeline_parallel.py index 9c90fe381bb2e..5346d67b10d16 100644 --- a/tests/distributed/test_pipeline_parallel.py +++ b/tests/distributed/test_pipeline_parallel.py @@ -185,7 +185,7 @@ TEXT_GENERATION_MODELS = { "mosaicml/mpt-7b": PPTestSettings.fast(), "nvidia/Minitron-8B-Base": PPTestSettings.fast(), "allenai/OLMo-1B-hf": PPTestSettings.fast(), - "shanearora/OLMo-7B-1124-hf": PPTestSettings.fast(), + "allenai/OLMo-2-0425-1B": PPTestSettings.fast(), "allenai/OLMoE-1B-7B-0924-Instruct": PPTestSettings.fast(), "facebook/opt-iml-max-1.3b": PPTestSettings.fast(), "OrionStarAI/Orion-14B-Chat": PPTestSettings.fast(), diff --git a/tests/models/registry.py b/tests/models/registry.py index 39b9795e7e166..84abd42e92313 100644 --- a/tests/models/registry.py +++ b/tests/models/registry.py @@ -205,7 +205,7 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { "MPTForCausalLM": _HfExamplesInfo("mosaicml/mpt-7b"), "NemotronForCausalLM": _HfExamplesInfo("nvidia/Minitron-8B-Base"), "OlmoForCausalLM": _HfExamplesInfo("allenai/OLMo-1B-hf"), - "Olmo2ForCausalLM": _HfExamplesInfo("shanearora/OLMo-7B-1124-hf"), + "Olmo2ForCausalLM": _HfExamplesInfo("allenai/OLMo-2-0425-1B"), "OlmoeForCausalLM": _HfExamplesInfo("allenai/OLMoE-1B-7B-0924-Instruct"), "OPTForCausalLM": _HfExamplesInfo("facebook/opt-125m", {"1b": "facebook/opt-iml-max-1.3b"}), diff --git a/vllm/test_utils.py b/vllm/test_utils.py index 8611a25922bb7..f8cec380f336e 100644 --- a/vllm/test_utils.py +++ b/vllm/test_utils.py @@ -110,7 +110,7 @@ MODELS_ON_S3 = [ "royokong/e5-v", "sentence-transformers/all-roberta-large-v1", "sentence-transformers/stsb-roberta-base-v2", - "shanearora/OLMo-7B-1124-hf", + "allenai/OLMo-2-0425-1B", "shuyuej/Llama-3.2-1B-Instruct-GPTQ", "ssmits/Qwen2-7B-Instruct-embed-base", "stabilityai/stablelm-3b-4e1t",