Skip to content

Commit 40d608e

Browse files
cccclaifacebook-github-bot
authored andcommitted
Add CI for conv_former and fastvit for QNN
Summary: #10191 fixes the model, try to enable them in CI Differential Revision: D73222005
1 parent f911567 commit 40d608e

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

.ci/scripts/test_model.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,10 @@ test_model_with_qnn() {
196196
EXPORT_SCRIPT=edsr
197197
# Additional deps for edsr
198198
pip install piq
199+
elif [[ "${MODEL_NAME}" == "fastvit" ]]; then
200+
EXPORT_SCRIPT=fastvit
201+
elif [[ "${MODEL_NAME}" == "conv_former" ]]; then
202+
EXPORT_SCRIPT=conv_former
199203
else
200204
echo "Unsupported model $MODEL_NAME"
201205
exit 1

.github/workflows/trunk.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -505,7 +505,7 @@ jobs:
505505
strategy:
506506
matrix:
507507
dtype: [fp32]
508-
model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l]
508+
model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l, fastvit, conv_former]
509509
fail-fast: false
510510
with:
511511
runner: linux.2xlarge

0 commit comments

Comments
 (0)