Pin pytorch version to 2.0.1 in order to avoid exit code 139 in Docker container (#512)

This commit is contained in:
sabaimran
2023-10-20 14:10:21 -07:00
committed by GitHub
parent e3f8a95784
commit 6dc0df3afb
2 changed files with 3 additions and 1 deletions

View File

@@ -52,7 +52,7 @@ dependencies = [
"schedule == 1.1.0",
"sentence-transformers == 2.2.2",
"transformers >= 4.28.0",
"torch >= 2.0.1",
"torch == 2.0.1",
"uvicorn == 0.17.6",
"aiohttp == 3.8.5",
"langchain >= 0.0.187",