We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0595584 commit ac35062Copy full SHA for ac35062
1 file changed
sdks/python/setup.py
@@ -571,7 +571,7 @@ def get_portability_package_data():
571
# in https://github.com/apache/beam/blob/master/sdks/python/tox.ini
572
# For more info, see
573
# https://docs.google.com/document/d/1c84Gc-cZRCfrU8f7kWGsNR2o8oSRjCM-dGHO9KvPWPw/edit?usp=sharing
574
- 'torch': ['torch<=1.13.0,<2.1.0'],
+ 'torch': ['torch>=1.9.0,<2.1.0'],
575
'tensorflow': ['tensorflow>=2.12rc1,<2.13'],
576
'transformers': [
577
'transformers>=4.28.0,<4.49.0',
0 commit comments