From e33e1623f6297fcd2929409fcbb62b246fa4dc57 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 24 Feb 2026 12:49:12 +0530 Subject: [PATCH] modify generate_model_tests.py --- utils/generate_model_tests.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/utils/generate_model_tests.py b/utils/generate_model_tests.py index 11acd2175e..c9729e29eb 100644 --- a/utils/generate_model_tests.py +++ b/utils/generate_model_tests.py @@ -72,6 +72,7 @@ OPTIONAL_TESTERS = [ # Other testers ("SingleFileTesterMixin", "single_file"), ("IPAdapterTesterMixin", "ip_adapter"), + ("AttentionBackendTesterMixin", "attention_backends"), ] @@ -530,6 +531,7 @@ def main(): "faster_cache", "single_file", "ip_adapter", + "attention_backends", "all", ], help="Optional testers to include",