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",