github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-21 18:53:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6362 from hpcaitech/CI/test_build_on_schedule

Browse Source 
[CI] Fix CI build on schedule error
This commit is contained in:
Hanks 2025-07-15 14:25:10 +08:00 committed by GitHub
commit edd65a84dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/build_on_schedule.yml vendored
View File

@ -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: |

9
tests/kit/model_zoo/timm/timm.py
View File

@ -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
)