From e285eb6993f4bf6fd280e221922c7fb278e5c06e Mon Sep 17 00:00:00 2001 From: botbw Date: Mon, 14 Jul 2025 02:38:02 +0000 Subject: [PATCH 1/2] [CI] install flash-attn 2.7.4.post1 --- .github/workflows/build_on_schedule.yml | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build_on_schedule.yml b/.github/workflows/build_on_schedule.yml index e71c8a84e..7038cf216 100644 --- a/.github/workflows/build_on_schedule.yml +++ b/.github/workflows/build_on_schedule.yml @@ -3,7 +3,7 @@ name: Build on Schedule on: schedule: # run at 00:00 of every Sunday - - cron: "0 0 * * *" + - cron: "0 0 * * 0" workflow_dispatch: jobs: @@ -51,6 +51,10 @@ jobs: with: ssh-key: ${{ secrets.SSH_KEY_FOR_CI }} + - name: Install flash-attention + run: | + pip install flash-attn==2.7.4.post1 --no-build-isolation + - name: Install Colossal-AI if: steps.check-avai.outputs.avai == 'true' run: | From 908c634686448d9e50534e1241b1c25c7cc14785 Mon Sep 17 00:00:00 2001 From: botbw Date: Mon, 14 Jul 2025 07:50:27 +0000 Subject: [PATCH 2/2] [CI] disable timm_regnetv_040 as aten::_unique2 is not supproted --- tests/kit/model_zoo/timm/timm.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/kit/model_zoo/timm/timm.py b/tests/kit/model_zoo/timm/timm.py index eb6d2f6bc..53b9385aa 100644 --- a/tests/kit/model_zoo/timm/timm.py +++ b/tests/kit/model_zoo/timm/timm.py @@ -119,9 +119,12 @@ model_zoo.register( model_zoo.register( name="timm_nf_regnet_b0", model_fn=tm.nf_regnet_b0, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn ) -model_zoo.register( - name="timm_regnetv_040", model_fn=tm.regnetv_040, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn -) + +# TODO: will need to register fake impl of aten::_unique2 to make it work (torch==2.5.1) +# model_zoo.register( +# name="timm_regnetv_040", model_fn=tm.regnetv_040, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn +# ) + model_zoo.register( name="timm_skresnet18", model_fn=tm.skresnet18, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn )