Add more requests in migration test

AlibabaPAI · Nov 20, 2024 · df144c4 · df144c4
1 parent 774f9d8
commit df144c4
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
@@ -96,7 +96,7 @@ async def run_bench_command(command):
 
     tasks = []
     for i in range(device_count//2):
-        bench_command = generate_bench_command(ip_ports=f"127.0.0.1:{base_port+i}", model=model, num_prompts=300,
+        bench_command = generate_bench_command(ip_ports=f"127.0.0.1:{base_port+i}", model=model, num_prompts=500,
                                                dataset_type="sharegpt",
                                                dataset_path="/mnt/dataset/sharegpt_gpt4/sharegpt_gpt4.jsonl" ,
                                                qps=10,