support v1 loader for machete (#3999)

2025-10-06 09:07:10 +08:00 · 2025-09-10 10:21:33 +08:00
parent b3fac5bde1
commit 3b1da6e4dd
2 changed files with 31 additions and 39 deletions
--- a/tests/model_loader/test_common_model.py
+++ b/tests/model_loader/test_common_model.py
@@ -72,6 +72,7 @@ model_param_map = {
                    "FD_ATTENTION_BACKEND": "MLA_ATTN",
                    "FLAGS_mla_use_tensorcore": "1",
                    "FLAGS_flash_attn_version": "3",
+                    "FD_USE_MACHETE": "1",
                },
            },
        ],