This is an automated email from the ASF dual-hosted git repository.
Amar3tto pushed a commit to branch fix-inference-ml
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/fix-inference-ml by this push:
new 1e736be50ab Pin torchvision
1e736be50ab is described below
commit 1e736be50ab65be0362187684238eccc9aa214c6
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Thu Aug 20 14:10:27 2026 +0400
Pin torchvision
---
.../workflows/beam_Inference_Python_Benchmarks_Dataflow.yml | 11 -----------
.../ml/inference/pytorch_image_captioning_requirements.txt | 3 ++-
2 files changed, 2 insertions(+), 12 deletions(-)
diff --git a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
index c8ef3830300..fc67923c6c2 100644
--- a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
+++ b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
@@ -107,17 +107,6 @@ jobs:
# The env variables are created and populated in the
test-arguments-action as
"<github.job>_test_arguments_<argument_file_paths_index>"
- name: get current time
run: echo "NOW_UTC=$(date '+%m%d%H%M%S' --utc)" >> $GITHUB_ENV
- - name: run PyTorch Image Object Detection Faster R-CNN ResNet-50 Batch
GPU
- uses: ./.github/actions/gradle-command-self-hosted-action
- timeout-minutes: 180
- with:
- gradle-command: :sdks:python:apache_beam:testing:load_tests:run
- arguments: |
-
-PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_image_object_detection_benchmarks
\
- -Prunner=DataflowRunner \
- -PpythonVersion=3.10 \
-
-PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_image_object_detection_requirements.txt
\
- '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_12 }} --device=GPU
--worker_accelerator=type=nvidia-tesla-t4,count=1,install-nvidia-driver=true
--sdk_location=container
--sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--mode=batch
--job_name=benchmark-tests-pytorch-image-object-detection-batch-${{env.NOW_UTC}}
--metrics_table=torch_inference_image_object_detection_batch_gpu
--output_table=apache-beam-t [...]
- name: run PyTorch Image Captioning BLIP + CLIP Batch GPU
uses: ./.github/actions/gradle-command-self-hosted-action
timeout-minutes: 180
diff --git
a/sdks/python/apache_beam/ml/inference/pytorch_image_captioning_requirements.txt
b/sdks/python/apache_beam/ml/inference/pytorch_image_captioning_requirements.txt
index 726cc147a0a..13a141865fc 100644
---
a/sdks/python/apache_beam/ml/inference/pytorch_image_captioning_requirements.txt
+++
b/sdks/python/apache_beam/ml/inference/pytorch_image_captioning_requirements.txt
@@ -15,7 +15,8 @@
# limitations under the License.
#
-torch>=2.6.0,<2.8.0
+torch==2.7.0
+torchvision==0.22.0
Pillow>=10.0.0
numpy>=1.25.0
transformers>=4.41.0,<5.0.0
