From 9b23d6dea2cfc404f4cd21900f8fe3a47106aff9 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 8 Apr 2025 07:59:19 -0700 Subject: [PATCH] Exclude naflex models from jit tests --- 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 3ba3615d..3a83b927 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -78,7 +78,7 @@ else: EXCLUDE_FILTERS = ['*enormous*'] NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*', '*_3b_*'] -EXCLUDE_JIT_FILTERS = ['hiera_*'] +EXCLUDE_JIT_FILTERS = ['hiera_*', '*naflex*'] TARGET_FWD_SIZE = MAX_FWD_SIZE = 384 TARGET_BWD_SIZE = 128