Skip to content

Commit 364f493

Browse files
authored
Fix qwen2.5 1.5b test (pytorch#14013)
Summary: Forward fix for pytorch#13972 Rollback Plan: Differential Revision: D81789757
1 parent 3899beb commit 364f493

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

.ci/scripts/test_model.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ test_model() {
9797
bash examples/models/llava/install_requirements.sh
9898
STRICT="--no-strict"
9999
fi
100-
if [[ "${MODEL_NAME}" == "qwen2_5" ]]; then
100+
if [[ "${MODEL_NAME}" == "qwen2_5_1_5b" ]]; then
101101
# Install requirements for export_llama
102102
bash examples/models/llama/install_requirements.sh
103103
# Test export_llm script: python3 -m extension.llm.export.export_llm.

.github/workflows/trunk.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ jobs:
176176
- model: phi_4_mini
177177
backend: portable
178178
runner: linux.arm64.m7g.4xlarge
179-
- model: qwen2_5
179+
- model: qwen2_5_1_5b
180180
backend: portable
181181
runner: linux.arm64.2xlarge
182182
- model: llama3_2_vision_encoder

examples/models/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ class Model(str, Enum):
3535
ResNet50 = "resnet50"
3636
Llava = "llava"
3737
EfficientSam = "efficient_sam"
38-
Qwen25 = "qwen2_5"
38+
Qwen25 = "qwen2_5_1_5b"
3939
Phi4Mini = "phi_4_mini"
4040
SmolLM2 = "smollm2"
4141

0 commit comments

Comments
 (0)