Add 3 models to 2 gpu runner in model downloading from nightly tests (#13261)

2025-11-13 22:59:18 -08:00
parent 9bc6a9adbe
commit 04848ba7cb
1 changed files with 3 additions and 0 deletions
--- a/scripts/ci/validate_and_download_models.py
+++ b/scripts/ci/validate_and_download_models.py
@@ -60,6 +60,9 @@ RUNNER_LABEL_MODEL_MAP: Dict[str, List[str]] = {
    "2-gpu-runner": [
        "mistralai/Mixtral-8x7B-Instruct-v0.1",
        "moonshotai/Kimi-Linear-48B-A3B-Instruct",
+        "Qwen/Qwen2-57B-A14B-Instruct",
+        "neuralmagic/Qwen2-72B-Instruct-FP8",
+        "zai-org/GLM-4.5-Air-FP8",
    ],
    "8-gpu-h200": [
        "deepseek-ai/DeepSeek-V3-0324",