|
53 | 53 | 'vision_transformer', 'vision_transformer_sam', 'vision_transformer_hybrid', 'vision_transformer_relpos',
|
54 | 54 | 'beit', 'mvitv2', 'eva', 'cait', 'xcit', 'volo', 'twins', 'deit', 'swin_transformer', 'swin_transformer_v2',
|
55 | 55 | 'swin_transformer_v2_cr', 'maxxvit', 'efficientnet', 'mobilenetv3', 'levit', 'efficientformer', 'resnet',
|
56 |
| - 'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit', 'hieradet_sam2' |
| 56 | + 'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit', 'hieradet_sam2', 'aimv2*' |
57 | 57 | ]
|
58 | 58 |
|
59 | 59 | # transformer / hybrid models don't support full set of spatial / feature APIs and/or have spatial output.
|
60 | 60 | NON_STD_FILTERS = [
|
61 | 61 | 'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
|
62 |
| - 'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*', |
| 62 | + 'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*', 'aimv2*', |
63 | 63 | 'poolformer_*', 'volo_*', 'sequencer2d_*', 'mvitv2*', 'gcvit*', 'efficientformer*', 'sam_hiera*',
|
64 | 64 | 'eva_*', 'flexivit*', 'eva02*', 'samvit_*', 'efficientvit_m*', 'tiny_vit_*', 'hiera_*', 'vitamin*', 'test_vit*',
|
65 | 65 | ]
|
|
73 | 73 | '*nfnet_f3*', '*nfnet_f4*', '*nfnet_f5*', '*nfnet_f6*', '*nfnet_f7*', '*efficientnetv2_xl*',
|
74 | 74 | '*resnetrs350*', '*resnetrs420*', 'xcit_large_24_p8*', '*huge*', '*giant*', '*gigantic*',
|
75 | 75 | '*enormous*', 'maxvit_xlarge*', 'regnet*1280', 'regnet*2560']
|
76 |
| - NON_STD_EXCLUDE_FILTERS = ['*huge*', '*giant*', '*gigantic*', '*enormous*'] |
| 76 | + NON_STD_EXCLUDE_FILTERS = ['*huge*', '*giant*', '*gigantic*', '*enormous*', '*_1b_*', '*_3b_*'] |
77 | 77 | else:
|
78 | 78 | EXCLUDE_FILTERS = ['*enormous*']
|
79 |
| - NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*'] |
| 79 | + NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*', '*_3b_*'] |
80 | 80 |
|
81 | 81 | EXCLUDE_JIT_FILTERS = ['hiera_*']
|
82 | 82 |
|
|
0 commit comments