From fd92ba0de85cb9da474a50d765601b36fee778ba Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 25 May 2021 12:52:07 -0700 Subject: [PATCH] Filter large vit models from torchscript tests --- tests/test_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 029ae0dd..de664068 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -178,7 +178,8 @@ if 'GITHUB_ACTIONS' not in os.environ: EXCLUDE_JIT_FILTERS = [ '*iabn*', 'tresnet*', # models using inplace abn unlikely to ever be scriptable - 'dla*', 'hrnet*', 'ghostnet*', # hopefully fix at some point + 'dla*', 'hrnet*', 'ghostnet*', # hopefully fix at some point + 'vit_large_*', 'vit_huge_*', ]