Update pyproject.toml

fix torch version
Desilo · Aug 6, 2024 · c176105 · c176105
1 parent a61e2b2
commit c176105
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -54,7 +54,7 @@ mpmath = "^1.3.0"
 scipy = "^1.10.1"
 matplotlib = "^3.7.1"
 joblib = "^1.2.0"
-torch = "^2.0.0"
+torch = "==2.1.1"
 tqdm = "^4.66.1"
 
 [tool.poetry.group.dev.dependencies]