Lock torch 1.12.0 for CPU

ahmetoner · Feb 5, 2023 · 881f12f · 881f12f
1 parent 70541ab
commit 881f12f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ python-multipart = "^0.0.5"
 ffmpeg-python = "^0.2.0"
 fastapi = "^0.85.0"
 fastapi-offline-swagger-ui = { git = "https://github.com/ahmetoner/fastapi-offline-swagger-ui" }
-torch = "1.12.0"
+torch = "=1.12.0"
 openai-whisper = { git = "https://github.com/openai/whisper.git", tag = "v20230124" }
 
 [tool.poetry.dev-dependencies]