diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 035d30f6adb..74eb75c6ddd 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -97,7 +97,7 @@ test_model() { bash examples/models/llava/install_requirements.sh STRICT="--no-strict" fi - if [[ "${MODEL_NAME}" == "qwen2_5" ]]; then + if [[ "${MODEL_NAME}" == "qwen2_5_1_5b" ]]; then # Install requirements for export_llama bash examples/models/llama/install_requirements.sh # Test export_llm script: python3 -m extension.llm.export.export_llm. diff --git a/.github/workflows/trunk.yml b/.github/workflows/trunk.yml index 3bbd0e48929..251bb238f1b 100644 --- a/.github/workflows/trunk.yml +++ b/.github/workflows/trunk.yml @@ -176,7 +176,7 @@ jobs: - model: phi_4_mini backend: portable runner: linux.arm64.m7g.4xlarge - - model: qwen2_5 + - model: qwen2_5_1_5b backend: portable runner: linux.arm64.2xlarge - model: llama3_2_vision_encoder diff --git a/examples/models/__init__.py b/examples/models/__init__.py index 82680a05c9d..c2212bbb218 100644 --- a/examples/models/__init__.py +++ b/examples/models/__init__.py @@ -35,7 +35,7 @@ class Model(str, Enum): ResNet50 = "resnet50" Llava = "llava" EfficientSam = "efficient_sam" - Qwen25 = "qwen2_5" + Qwen25 = "qwen2_5_1_5b" Phi4Mini = "phi_4_mini" SmolLM2 = "smollm2"