diff --git a/sdks/python/apache_beam/examples/inference/runinference_metrics/setup.py b/sdks/python/apache_beam/examples/inference/runinference_metrics/setup.py index fd2c07cc4e1b..5f492f23fad9 100644 --- a/sdks/python/apache_beam/examples/inference/runinference_metrics/setup.py +++ b/sdks/python/apache_beam/examples/inference/runinference_metrics/setup.py @@ -29,7 +29,7 @@ from setuptools import find_packages REQUIREMENTS = [ - "apache-beam[gcp]==2.41.0", "transformers==4.38.0", "torch==1.13.1" + "apache-beam[gcp]==2.41.0", "transformers==4.38.0", "torch==2.13.0" ] setuptools.setup(