From bce8400aa7a17ae51fe28e2d448f1dc292a7f039 Mon Sep 17 00:00:00 2001 From: Chen Lai Date: Fri, 8 Aug 2025 14:05:27 -0700 Subject: [PATCH] Enable fbnet model in CI (#13201) Summary: Rollback Plan: Differential Revision: D79741919 --- .ci/scripts/test_model.sh | 6 +++++- .github/workflows/trunk.yml | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 035d30f6adb..5a2c5661532 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -211,6 +211,10 @@ test_model_with_qnn() { EXPORT_SCRIPT=efficientnet elif [[ "${MODEL_NAME}" == "eurobert" ]]; then EXPORT_SCRIPT=eurobert + elif [[ "${MODEL_NAME}" == "fbnet" ]]; then + pip install timm + EXPORT_SCRIPT=fbnet + EXTRA_FLAGS="--dataset imagenet-mini/val" elif [[ "${MODEL_NAME}" == "focalnet" ]]; then EXPORT_SCRIPT=focalnet elif [[ "${MODEL_NAME}" == "mobilevit_v1" ]]; then @@ -240,7 +244,7 @@ test_model_with_qnn() { "cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin") SCRIPT_FOLDER=oss_scripts ;; - "albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1") + "albert"|"bert"|"conv_former"|"distilbert"|"fbnet"|"roberta"|"efficientnet"|"mobilevit_v1") pip install evaluate SCRIPT_FOLDER=oss_scripts # 16bit models will encounter op validation fail on some operations, diff --git a/.github/workflows/trunk.yml b/.github/workflows/trunk.yml index fc2cb36cccb..44288018258 100644 --- a/.github/workflows/trunk.yml +++ b/.github/workflows/trunk.yml @@ -648,7 +648,7 @@ jobs: strategy: matrix: dtype: [fp32] - model: [cvt, dit, efficientnet, focalnet, mobilevit_v1, mobilevit_v2, pvt, swin, albert, bert, distilbert, roberta] # eurobert requires transfomer >= 4.48.0, skip for now + 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 fail-fast: false with: runner: linux.2xlarge