You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by vt...@apache.org on 2026/04/16 13:22:24 UTC
(beam) 01/01: Fix configs
This is an automated email from the ASF dual-hosted git repository.
vterentev pushed a commit to branch fix-inference-tests
in repository https://gitbox.apache.org/repos/asf/beam.git
commit d9eb9d6365a51cea3c8e1ffe00ffba54a330dae8
Author: Vitaly Terentyev <vi...@akvelon.com>
AuthorDate: Thu Apr 16 17:22:07 2026 +0400
Fix configs
---
...rks_Dataflow_Pytorch_Sentiment_Batch_DistilBert_Base_Uncased.txt | 6 ++++--
...Dataflow_Pytorch_Sentiment_Streaming_DistilBert_Base_Uncased.txt | 3 ++-
2 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Batch_DistilBert_Base_Uncased.txt b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Batch_DistilBert_Base_Uncased.txt
index 4642156d57c..74377880c90 100644
--- a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Batch_DistilBert_Base_Uncased.txt
+++ b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Batch_DistilBert_Base_Uncased.txt
@@ -28,8 +28,10 @@
--metrics_table=torch_sentiment_distilbert_base_uncased_batch
--input_options={}
--influx_measurement=torch_sentiment_distilbert_base_uncased
---device=CPU
+--device=GPU
--input_file=gs://apache-beam-ml/testing/inputs/sentences_50k.txt
--runner=DataflowRunner
--model_path=distilbert-base-uncased-finetuned-sst-2-english
---model_state_dict_path=gs://apache-beam-ml/models/huggingface.sentiment.distilbert-base-uncased.pth
\ No newline at end of file
+--experiments=use_runner_v2
+--dataflow_service_options=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver
+--model_state_dict_path=gs://apache-beam-ml/models/huggingface.sentiment.distilbert-base-uncased.pth
diff --git a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Streaming_DistilBert_Base_Uncased.txt b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Streaming_DistilBert_Base_Uncased.txt
index d10b9bb2dfc..5f9e2ce677f 100644
--- a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Streaming_DistilBert_Base_Uncased.txt
+++ b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Sentiment_Streaming_DistilBert_Base_Uncased.txt
@@ -28,9 +28,10 @@
--metrics_table=torch_sentiment_distilbert_base_uncased
--input_options={}
--influx_measurement=torch_sentiment_distilbert_base_uncased
---device=CPU
+--device=GPU
--input_file=gs://apache-beam-ml/testing/inputs/sentences_50k.txt
--runner=DataflowRunner
+--experiments=use_runner_v2
--dataflow_service_options=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver
--model_path=distilbert-base-uncased-finetuned-sst-2-english
--model_state_dict_path=gs://apache-beam-ml/models/huggingface.sentiment.distilbert-base-uncased.pth