github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-24 11:08:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CI] disable timm_regnetv_040 as aten::_unique2 is not supproted

Browse Source
This commit is contained in:
botbw
2025-07-14 07:50:27 +00:00
parent e285eb6993
commit 908c634686
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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
)