From 58f8f4e4086cae43a67de210e56de4af10e8b066 Mon Sep 17 00:00:00 2001 From: Kangyan-Zhou Date: Mon, 17 Nov 2025 13:10:37 -0800 Subject: [PATCH] Add missing models (#13456) --- .github/workflows/pr-test.yml | 8 -------- scripts/ci/validate_and_download_models.py | 7 +++++++ 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 48662d565..3cb8f8ced 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -320,14 +320,6 @@ jobs: run: | CUSTOM_BUILD_SGL_KERNEL=${{needs.check-changes.outputs.sgl_kernel}} bash scripts/ci/ci_install_dependency.sh diffusion - - name: Clean Corrupted Hugging Face Model Cache - run: | - echo "Temp: Deleting potentially corrupted Qwen/Qwen-Image and Qwen/Qwen-Image-Edit cache to ensure a fresh download. This is temporary" - rm -rf /hf_home/hub/models--Qwen--Qwen-Image - rm -rf /hf_home/hub/models--Qwen--Qwen-Image-Edit - rm -rf /hf_home/hub/models--Wan-AI--Wan2.2-I2V-A14B-Diffusers - rm -rf /hf_home/hub/models--Wan-AI--Wan2.2-TI2V-5B-Diffusers - - name: Run diffusion server tests timeout-minutes: 60 run: | diff --git a/scripts/ci/validate_and_download_models.py b/scripts/ci/validate_and_download_models.py index f1ae62c56..d1ef0f4e0 100755 --- a/scripts/ci/validate_and_download_models.py +++ b/scripts/ci/validate_and_download_models.py @@ -46,6 +46,7 @@ RUNNER_LABEL_MODEL_MAP: Dict[str, List[str]] = { "lmsys/sglang-ci-dsv3-test", "lmsys/sglang-EAGLE-llama2-chat-7B", "lmsys/sglang-EAGLE3-LLaMA3.1-Instruct-8B", + "LxzGordon/URM-LLaMa-3.1-8B", "marco/mcdse-2b-v1", "meta-llama/Llama-2-7b-chat-hf", "meta-llama/Llama-3.2-1B-Instruct", @@ -54,6 +55,7 @@ RUNNER_LABEL_MODEL_MAP: Dict[str, List[str]] = { "moonshotai/Kimi-VL-A3B-Instruct", "nvidia/NVIDIA-Nemotron-Nano-9B-v2", "nvidia/NVIDIA-Nemotron-Nano-9B-v2-FP8", + "openai/gpt-oss-20b", "OpenGVLab/InternVL2_5-2B", "Qwen/Qwen2.5-7B-Instruct", "Qwen/Qwen3-8B", @@ -61,6 +63,11 @@ RUNNER_LABEL_MODEL_MAP: Dict[str, List[str]] = { "Qwen/Qwen3-Embedding-8B", "Qwen/QwQ-32B-AWQ", "Qwen/Qwen3-30B-A3B", + "Qwen/Qwen-Image", + "Qwen/Qwen-Image-Edit", + "Skywork/Skywork-Reward-Llama-3.1-8B-v0.2", + "Wan-AI/Wan2.2-I2V-A14B-Diffusers", + "Wan-AI/Wan2.2-TI2V-5B-Diffusers", ], "2-gpu-runner": [ "mistralai/Mixtral-8x7B-Instruct-v0.1",