[CI] Tune threshold (#5787)

2025-04-27 04:10:22 -07:00
parent 9c088829ee
commit 3c4e0ee64d
3 changed files with 5 additions and 5 deletions
--- a/test/srt/test_bench_one_batch.py
+++ b/test/srt/test_bench_one_batch.py
@@ -51,7 +51,7 @@ class TestBenchOneBatch(CustomTestCase):
                f"### test_torch_compile_tp2_bs1\n"
                f"output_throughput : {output_throughput:.2f} token/s\n"
            )
-            self.assertGreater(output_throughput, 235)
+            self.assertGreater(output_throughput, 225)


 if __name__ == "__main__":