Fix versions

microsoft · Aug 6, 2024 · a24bbde · a24bbde
1 parent 947ca7c
commit a24bbde
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -32,9 +32,9 @@ dynamic = ["version"]
 
 requires-python = ">=3.10"
 dependencies = [
-    "numpy",
-    "torch",
-    "einops",
+    "numpy==1.26.2",
+    "torch==2.2.0",
+    "einops==0.6.1",
     "timm==0.6.13",
     "huggingface-hub",
 ]
@@ -50,7 +50,6 @@ dev = [
     "ruff==0.4.1",
     "pre-commit",
     "jupyter-book",
-    "torch==2.2.0",
 ]
 
 [project.urls]