Merge pull request #1894 from seefun/master

add two different EfficientViT models
huggingface · Aug 19, 2023 · b801156 · b801156
2 parents 24fc720 + 7d7589e
commit b801156
Show file tree

Hide file tree

Showing 4 changed files with 1,332 additions and 1 deletion.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -41,7 +41,7 @@
     'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
     'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*', 'beit*',
     'poolformer_*', 'volo_*', 'sequencer2d_*', 'pvt_v2*', 'mvitv2*', 'gcvit*', 'efficientformer*',
-    'eva_*', 'flexivit*', 'eva02*', 'samvit_*'
+    'eva_*', 'flexivit*', 'eva02*', 'samvit_*', 'efficientvit_m*'
 ]
 NUM_NON_STD = len(NON_STD_FILTERS)
 

diff --git a/timm/models/__init__.py b/timm/models/__init__.py
@@ -17,6 +17,8 @@
 from .efficientformer import *
 from .efficientformer_v2 import *
 from .efficientnet import *
+from .efficientvit_mit import *
+from .efficientvit_msra import *
 from .eva import *
 from .focalnet import *
 from .gcvit import *