Add missing models (#13456)

This commit is contained in:
Kangyan-Zhou
2025-11-17 13:10:37 -08:00
committed by GitHub
parent 25acbbc612
commit 58f8f4e408
2 changed files with 7 additions and 8 deletions

View File

@@ -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: |

View File

@@ -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",