Update torch version to 1.13 because of the weights_only parameter in…

… torch.load Also, update tomli version requirements
stanfordnlp · Dec 30, 2024 · 59117ba · 59117ba
1 parent 34d7c8d
commit 59117ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -84,8 +84,8 @@
         'protobuf>=3.15.0',
         'requests', 
         'networkx',
-        'tomli;python_version<"3.11"',
-        'torch>=1.6.0',
+        'tomli;python_version<"3.12"',
+        'torch>=1.13.0',
         'tqdm',
     ],