Skip to content

Commit bce8400

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

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
@@ -211,6 +211,10 @@ test_model_with_qnn() {
211211
EXPORT_SCRIPT=efficientnet
212212
elif [[ "${MODEL_NAME}" == "eurobert" ]]; then
213213
EXPORT_SCRIPT=eurobert
214+
elif [[ "${MODEL_NAME}" == "fbnet" ]]; then
215+
pip install timm
216+
EXPORT_SCRIPT=fbnet
217+
EXTRA_FLAGS="--dataset imagenet-mini/val"
214218
elif [[ "${MODEL_NAME}" == "focalnet" ]]; then
215219
EXPORT_SCRIPT=focalnet
216220
elif [[ "${MODEL_NAME}" == "mobilevit_v1" ]]; then
@@ -240,7 +244,7 @@ test_model_with_qnn() {
240244
"cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin")
241245
SCRIPT_FOLDER=oss_scripts
242246
;;
243-
"albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
247+
"albert"|"bert"|"conv_former"|"distilbert"|"fbnet"|"roberta"|"efficientnet"|"mobilevit_v1")
244248
pip install evaluate
245249
SCRIPT_FOLDER=oss_scripts
246250
# 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
@@ -648,7 +648,7 @@ jobs:
648648
strategy:
649649
matrix:
650650
dtype: [fp32]
651-
model: [cvt, dit, efficientnet, focalnet, mobilevit_v1, mobilevit_v2, pvt, swin, albert, bert, distilbert, roberta] # eurobert requires transfomer >= 4.48.0, skip for now
651+
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
652652
fail-fast: false
653653
with:
654654
runner: linux.2xlarge

0 commit comments

Comments
 (0)