modify generate_model_tests.py

2026-02-24 20:01:34 +08:00 · 2026-02-24 12:49:12 +05:30
parent 6648604f0d
commit e33e1623f6
1 changed files with 2 additions and 0 deletions
--- 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",