Add missing models (#13456)
This commit is contained in:
8
.github/workflows/pr-test.yml
vendored
8
.github/workflows/pr-test.yml
vendored
@@ -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: |
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user