From 7ba64e632c38426831614c270e66ea13e136499a Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Thu, 1 Feb 2024 10:13:15 -0500 Subject: [PATCH] Revert "[don't merge yet] unpin torch (#2406)" (#2407) This reverts commit 8b770a7dabd957ae54f1abb028d1ce53db6cf4d4. --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index ef8263c671d..4e325df1a36 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ ] }, python_requires=">=3.8.0", - install_requires=["numpy>=1.17", "packaging>=20.0", "psutil", "pyyaml", "torch>=1.10.0", "huggingface_hub", "safetensors>=0.3.1"], + install_requires=["numpy>=1.17", "packaging>=20.0", "psutil", "pyyaml", "torch>=1.10.0,<2.2.0", "huggingface_hub", "safetensors>=0.3.1"], extras_require=extras, classifiers=[ "Development Status :: 5 - Production/Stable",