diff --git a/setup.py b/setup.py index a97a99e..567f234 100644 --- a/setup.py +++ b/setup.py @@ -24,7 +24,7 @@ "tqdm>=4.62.3", "setuptools>=65.5.1", # Needed to install dynamic packages from source (e.g. Blink) "prettytable>=3.4", - "torch>=1,<2", + "torch>=1", "transformers>=4.20", "datasets>=2.9.1", "evaluate>=0.3.0",