Skip to content

Commit e33e162

Browse files
committed
modify generate_model_tests.py
1 parent 6648604 commit e33e162

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

utils/generate_model_tests.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@
7272
# Other testers
7373
("SingleFileTesterMixin", "single_file"),
7474
("IPAdapterTesterMixin", "ip_adapter"),
75+
("AttentionBackendTesterMixin", "attention_backends"),
7576
]
7677

7778

@@ -530,6 +531,7 @@ def main():
530531
"faster_cache",
531532
"single_file",
532533
"ip_adapter",
534+
"attention_backends",
533535
"all",
534536
],
535537
help="Optional testers to include",

0 commit comments

Comments
 (0)