diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 6c799ba943bb..a0bbc301435b 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -571,7 +571,7 @@ def get_portability_package_data(): # in https://github.com/apache/beam/blob/master/sdks/python/tox.ini # For more info, see # https://docs.google.com/document/d/1c84Gc-cZRCfrU8f7kWGsNR2o8oSRjCM-dGHO9KvPWPw/edit?usp=sharing - 'torch': ['torch<=1.13.0,<2.1.0'], + 'torch': ['torch>=1.9.0,<2.1.0'], 'tensorflow': ['tensorflow>=2.12rc1,<2.13'], 'transformers': [ 'transformers>=4.28.0,<4.49.0',