From fe315331cc655ac6da9a52d260433122835a475e Mon Sep 17 00:00:00 2001
From: Chen Lai <chenlai@meta.com>
Date: Fri, 5 Sep 2025 09:31:30 -0700
Subject: [PATCH] Fix qwen2.5 1.5b test

Summary:
Forward fix for https://github.com/pytorch/executorch/pull/13972

Rollback Plan:

Differential Revision: D81789757
---
 .ci/scripts/test_model.sh   | 2 +-
 .github/workflows/trunk.yml | 2 +-
 examples/models/__init__.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

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"