⬆️ Update torch requirements (#71)

Signed-off-by: Marcos Martinez <Marcos.Martinez.Galindo@ibm.com>
IBM · Oct 4, 2023 · 08f47f3 · 08f47f3
1 parent f98a878
commit 08f47f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",