Change from 500 to 300

AlibabaPAI · Dec 5, 2024 · 62266c8 · 62266c8
1 parent b739dfb
commit 62266c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e_test/test_migration.py b/tests/e2e_test/test_migration.py
@@ -116,7 +116,7 @@ def run_bench_command(command):
         bench_command = generate_bench_command(
             ip_ports=f"127.0.0.1:{base_port + i}",
             model=model,
-            num_prompts=500,
+            num_prompts=300,
             dataset_type="sharegpt",
             dataset_path="/mnt/dataset/sharegpt_gpt4/sharegpt_gpt4.jsonl",
             qps=10,