From 9de6873bdece5e65cc258c02b8747d55c729ad1a Mon Sep 17 00:00:00 2001 From: wilhelmagren Date: Sun, 19 Nov 2023 15:56:40 +0100 Subject: [PATCH] [fix] change torch cpu repo name --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index d541994..06fa543 100644 --- 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"