fix: lock torch version to 2.0.0

winstxnhdw · Oct 4, 2023 · 11ea300 · 11ea300
1 parent 4ae404a
commit 11ea300
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,7 +45,7 @@ pymupdf = "^1.23.4"
 [tool.poetry.group.special.dependencies]
 transformers = { git = "https://github.com/huggingface/transformers.git", rev = "main" }
 redis = { extras = ["hiredis"], version = "^5.0.1" }
-torch = ">=2.0.0, !=2.0.1"
+torch = "2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 pyright = "^1.1.328"