Skip to content

Commit cbd1b14

Browse files
cccclaifacebook-github-bot
authored andcommitted
Enable fbnet model in CI (#13201)
Summary: Rollback Plan: Differential Revision: D79741919
1 parent afdbb85 commit cbd1b14

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

.ci/scripts/test_model.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,10 @@ test_model_with_qnn() {
212212
EXPORT_SCRIPT=efficientnet
213213
elif [[ "${MODEL_NAME}" == "eurobert" ]]; then
214214
EXPORT_SCRIPT=eurobert
215+
elif [[ "${MODEL_NAME}" == "fbnet" ]]; then
216+
pip install timm==0.1.24
217+
EXPORT_SCRIPT=fbnet
218+
EXTRA_FLAGS="--dataset imagenet-mini/val"
215219
elif [[ "${MODEL_NAME}" == "focalnet" ]]; then
216220
EXPORT_SCRIPT=focalnet
217221
elif [[ "${MODEL_NAME}" == "mobilevit_v1" ]]; then
@@ -241,7 +245,7 @@ test_model_with_qnn() {
241245
"cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin")
242246
SCRIPT_FOLDER=oss_scripts
243247
;;
244-
"albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
248+
"albert"|"bert"|"conv_former"|"distilbert"|"fbnet"|"roberta"|"efficientnet"|"mobilevit_v1")
245249
pip install evaluate
246250
SCRIPT_FOLDER=oss_scripts
247251
# 16bit models will encounter op validation fail on some operations,

.github/workflows/trunk.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,7 @@ jobs:
628628
strategy:
629629
matrix:
630630
dtype: [fp32]
631-
model: [cvt, dit, efficientnet, focalnet, mobilevit_v1, mobilevit_v2, pvt, swin, albert, bert, distilbert, roberta] # eurobert requires transfomer >= 4.48.0, skip for now
631+
model: [cvt, dit, efficientnet, focalnet, mobilevit_v1, mobilevit_v2, pvt, swin, albert, bert, distilbert, fbnet, roberta] # eurobert requires transfomer >= 4.48.0, skip for now
632632
fail-fast: false
633633
with:
634634
runner: linux.2xlarge

0 commit comments

Comments
 (0)