Add fastvit to forward_intermediates test

rwightman · rwightman · commit ce637771dc2a · 2024-05-30T10:18:24.000-07:00
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -52,7 +52,7 @@
     'vision_transformer', 'vision_transformer_sam', 'vision_transformer_hybrid', 'vision_transformer_relpos',
     'beit', 'mvitv2', 'eva', 'cait', 'xcit', 'volo', 'twins', 'deit', 'swin_transformer', 'swin_transformer_v2',
     'swin_transformer_v2_cr', 'maxxvit', 'efficientnet', 'mobilenetv3', 'levit', 'efficientformer', 'resnet',
-    'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera',
+    'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit',
 ]
 
 # transformer / hybrid models don't support full set of spatial / feature APIs and/or have spatial output.

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@`
`52`	`52`	`'vision_transformer', 'vision_transformer_sam', 'vision_transformer_hybrid', 'vision_transformer_relpos',`
`53`	`53`	`'beit', 'mvitv2', 'eva', 'cait', 'xcit', 'volo', 'twins', 'deit', 'swin_transformer', 'swin_transformer_v2',`
`54`	`54`	`'swin_transformer_v2_cr', 'maxxvit', 'efficientnet', 'mobilenetv3', 'levit', 'efficientformer', 'resnet',`
`55`		`- 'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera',`
	`55`	`+ 'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit',`
`56`	`56`	`]`
`57`	`57`
`58`	`58`	`# transformer / hybrid models don't support full set of spatial / feature APIs and/or have spatial output.`