[fix] change torch cpu repo name

neurocode-ai · Nov 19, 2023 · 9de6873 · 9de6873
1 parent 38323a2
commit 9de6873
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,7 +43,7 @@ matplotlib = "^3.8.2"
 tqdm = "^4.66.1"
 scikit-learn = "^1.3.2"
 pillow = "^10.1.0"
-torch = { version = "2.0.1+cpu", source = "torch_cpu" }
+torch = { version = "2.0.1+cpu", source = "torch-cpu" }
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.11.0"