From ce637771dc2abc5ae81a96af14724300815edd15 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 30 May 2024 10:18:24 -0700 Subject: [PATCH] Add fastvit to forward_intermediates test --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 652ea355..7d9c041e 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -52,7 +52,7 @@ FEAT_INTER_FILTERS = [ '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.