address PR code review

2026-03-22 17:23:12 +00:00
parent 0a566d7a15
commit 6636e5de7e
9 changed files with 51 additions and 17 deletions
--- a/tests/e2e/test_llama4_moe_aux_free.py
+++ b/tests/e2e/test_llama4_moe_aux_free.py
@@ -21,7 +21,7 @@ class TestLlama4MoeAuxFree(unittest.TestCase):
            {
                "base_model": "yujiepan/llama-4-tiny-random",
                "tokenizer_config": "yujiepan/llama-4-tiny-random",
-                "trust_remote_code": True,
+                "trust_remote_code": False,
                "flash_attention": False,
                "sequence_len": 512,
                "bf16": False,