Add FlashInternImage models

tests/test_models.py

@@ -484,7 +484,7 @@ def _create_fx_model(model, train=False):
     return fx_model
 
 
-EXCLUDE_FX_FILTERS = ['vit_gi*']
+EXCLUDE_FX_FILTERS = ['vit_gi*', '*flash_intern_image*']
 # not enough memory to run fx on more models than other tests
 if 'GITHUB_ACTIONS' in os.environ:
     EXCLUDE_FX_FILTERS += [

timm/models/__init__.py

@@ -21,6 +21,7 @@
 from .efficientvit_msra import *
 from .eva import *
 from .fastvit import *
+from .flash_intern_image import *
 from .focalnet import *
 from .gcvit import *
 from .ghostnet import *