diff --git a/test/registered/mla/test_mla_deepseek_v3.py b/test/registered/mla/test_mla_deepseek_v3.py index 86c9fef79..85236be24 100644 --- a/test/registered/mla/test_mla_deepseek_v3.py +++ b/test/registered/mla/test_mla_deepseek_v3.py @@ -144,7 +144,7 @@ class TestMLADeepseekV3Fa3Fp8Kvcache(CustomTestCase): metrics = run_eval_few_shot_gsm8k(args) print(metrics) - self.assertGreater(metrics["accuracy"], 0.62) + self.assertGreater(metrics["accuracy"], 0.60) class TestDeepseekV3MTP(CustomTestCase): diff --git a/test/registered/quant/test_autoround.py b/test/registered/quant/test_autoround.py index c9d07cd93..f2baf0617 100644 --- a/test/registered/quant/test_autoround.py +++ b/test/registered/quant/test_autoround.py @@ -55,7 +55,7 @@ class TestAutoRound(CustomTestCase): if "Llama" in model: self.assertGreaterEqual(metrics["score"], 0.6) else: - self.assertGreaterEqual(metrics["score"], 0.26) + self.assertGreaterEqual(metrics["score"], 0.25) finally: kill_process_tree(process.pid) print(f"[INFO] Server for {model} stopped.")