Skip to content

Commit 5f0ebdf

Browse files
committed
200 gb disk
1 parent e3d2101 commit 5f0ebdf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_VLLM_Gemma_Batch.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,9 @@
1818
--temp_location=gs://temp-storage-for-perf-tests/loadtests
1919
--staging_location=gs://temp-storage-for-perf-tests/loadtests
2020
--input=gs://apache-beam-ml/testing/inputs/sentences_50k.txt
21-
--machine_type=g2-standard-8
21+
--machine_type=n1-standard-8
2222
--worker_zone=us-central1-b
23-
--disk_size_gb=80
23+
--disk_size_gb=200
2424
--input_options={}
2525
--num_workers=8
2626
--max_num_workers=25
@@ -32,5 +32,5 @@
3232
--metrics_table=gemma_vllm_batch
3333
--influx_measurement=gemma_vllm_batch
3434
--model_gcs_path=gs://apache-beam-ml/models/gemma-2b-it
35-
--dataflow_service_options=worker_accelerator=type:nvidia-l4;count:1;install-nvidia-driver
35+
--dataflow_service_options=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver
3636
--experiments=use_runner_v2

0 commit comments

Comments
 (0)