bump torch

clementchadebec · Oct 22, 2023 · 8b78611 · 8b78611
1 parent 5e19262
commit 8b78611
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ numpy>=1.19
 pydantic>=2.0
 scikit-learn
 scipy>=1.7.1
-torch>=1.10.1
+torch>=2.0
 tqdm
 typing_extensions
 dataclasses>=0.6

diff --git a/setup.py b/setup.py
@@ -32,7 +32,7 @@
  "pydantic>=2.0",
  "scikit-learn",
  "scipy>=1.7.1",
- "torch>=1.10.1",
+ "torch>=2.0",
  "tqdm",
  "typing_extensions",
  "dataclasses>=0.6",