Skip to content

Commit 908c634

Browse files
committed
[CI] disable timm_regnetv_040 as aten::_unique2 is not supproted
1 parent e285eb6 commit 908c634

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/kit/model_zoo/timm/timm.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,9 +119,12 @@
119119
model_zoo.register(
120120
name="timm_nf_regnet_b0", model_fn=tm.nf_regnet_b0, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn
121121
)
122-
model_zoo.register(
123-
name="timm_regnetv_040", model_fn=tm.regnetv_040, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn
124-
)
122+
123+
# TODO: will need to register fake impl of aten::_unique2 to make it work (torch==2.5.1)
124+
# model_zoo.register(
125+
# name="timm_regnetv_040", model_fn=tm.regnetv_040, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn
126+
# )
127+
125128
model_zoo.register(
126129
name="timm_skresnet18", model_fn=tm.skresnet18, data_gen_fn=data_gen_fn, output_transform_fn=output_transform_fn
127130
)

0 commit comments

Comments
 (0)