bump torch ver. 1.7->1.8

445beade · haileyschoelkopf · 46ca66de · 445beade
Commit 445beade authored Jul 03, 2023 by haileyschoelkopf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@ setuptools.setup(
        "sacrebleu==1.5.0",
        "scikit-learn>=0.24.1",
        "sqlitedict",
-        "torch>=1.7",
+        "torch>=1.8",
        "tqdm-multiprocess",
        "transformers>=4.1",
        "zstandard",